🧠 Neural network API

API Documentation

• Activations
  • CeLU
  • ELU
  • GELU
  • GLU
  • HardShrink
  • HardSigmoid
  • HardSwish
  • HardTanh
  • LeakyReLU
  • LogSigmoid
  • LogSoftmax
  • Mish
  • PReLU
  • ReLU
  • ReLU6
  • SeLU
  • Sigmoid
  • SoftPlus
  • SoftShrink
  • SoftSign
  • SquarePlus
  • Swish
  • Tanh
  • TanhShrink
  • ThresholdedReLU
• Attention
  • MultiHeadAttention
• Convolution
  • Conv1D
  • Conv2D
  • Conv3D
  • Conv1DTranspose
  • Conv2DTranspose
  • Conv3DTranspose
  • DepthwiseConv1D
  • DepthwiseConv2D
  • DepthwiseConv3D
  • SeparableConv1D
  • SeparableConv2D
  • SeparableConv3D
  • Conv1DLocal
  • Conv2DLocal
  • Conv3DLocal
  • FFTConv1D
  • FFTConv2D
  • FFTConv3D
  • DepthwiseFFTConv1D
  • DepthwiseFFTConv2D
  • DepthwiseFFTConv3D
  • FFTConv1DTranspose
  • FFTConv2DTranspose
  • FFTConv3DTranspose
  • SeparableFFTConv1D
  • SeparableFFTConv2D
  • SeparableFFTConv3D
  • SpectralConv1D
  • SpectralConv2D
  • SpectralConv3D
  • conv_nd()
  • depthwise_conv_nd()
  • depthwise_fft_conv_nd()
  • fft_conv_nd()
  • local_conv_nd()
  • separable_conv_nd()
  • separable_fft_conv_nd()
  • conv_nd_transpose()
  • fft_conv_nd_transpose()
  • spectral_conv_nd()
• Dropout
  • Dropout
  • Dropout1D
  • Dropout2D
  • Dropout3D
  • RandomCutout1D
  • RandomCutout2D
  • RandomCutout3D
  • dropout_nd()
  • random_cutout_nd()
• Linear
  • Linear
  • Identity
  • Embedding
  • MLP
  • linear()
• Normalization
  • LayerNorm
  • InstanceNorm
  • GroupNorm
  • BatchNorm
  • EvalBatchNorm
  • layer_norm()
  • instance_norm()
  • group_norm()
  • batch_norm()
  • eval_batch_norm()
  • weight_norm()
• Pooling
  • MaxPool1D
  • MaxPool2D
  • MaxPool3D
  • AvgPool1D
  • AvgPool2D
  • AvgPool3D
  • GlobalAvgPool1D
  • GlobalAvgPool2D
  • GlobalAvgPool3D
  • GlobalMaxPool1D
  • GlobalMaxPool2D
  • GlobalMaxPool3D
  • LPPool1D
  • LPPool2D
  • LPPool3D
  • AdaptiveAvgPool1D
  • AdaptiveAvgPool2D
  • AdaptiveAvgPool3D
  • AdaptiveMaxPool1D
  • AdaptiveMaxPool2D
  • AdaptiveMaxPool3D
  • adaptive_avg_pool_nd()
  • adaptive_max_pool_nd()
  • avg_pool_nd()
  • lp_pool_nd()
  • max_pool_nd()
• Recurrent
  • LSTMCell
  • GRUCell
  • SimpleRNNCell
  • LinearCell
  • ConvLSTM1DCell
  • ConvLSTM2DCell
  • ConvLSTM3DCell
  • ConvGRU1DCell
  • ConvGRU2DCell
  • ConvGRU3DCell
  • FFTConvLSTM1DCell
  • FFTConvLSTM2DCell
  • FFTConvLSTM3DCell
  • FFTConvGRU1DCell
  • FFTConvGRU2DCell
  • FFTConvGRU3DCell
  • scan_cell()
• Reshaping
  • CenterCrop1D
  • CenterCrop2D
  • CenterCrop3D
  • RandomCrop1D
  • RandomCrop2D
  • RandomCrop3D
  • Upsample1D
  • Upsample2D
  • Upsample3D
  • center_crop_nd()
  • extract_patches()
  • random_crop_nd()
  • upsample_nd()