Conv1d
Conv2d
Conv3d
ConvTranpose1d
ConvTranpose2d
ConvTranpose3d
LazyConv1d
LazyConv2d
LazyConv3d
LazyConvTranspose1d
LazyConvTranspose2d
LazyConvTranspose3d
Unfold
Fold
MaxPool1d
MaxPool2d
MaxPool3d
MaxUnPool1d
MaxUnPool2d
MaxUnPool3d
AvgPool1d
AvgPool2d
AvgPool3d
FractionalMaxPool2d
FractionalMaxPool3d
LPPool1d
LPPool2d
AdaptiveMaxPool1d
AdaptiveMaxPool2d
AdaptiveMaxPool3d
AdaptiveAvgPool1d
AdaptiveAvgPool2d
AdaptiveAvgPool3d
ReflectionPad1d
ReflectionPad2d
ReflectionPad3d
ReplicationPad1d
ReplicationPad2d
ReplicationPad3d
ZeroPad2d
ConstantPad1d
ConstantPad2d
ConstantPad3d
MultiheadAttention
ELU
Hardshrink
Hardsigmoid
Hardtanh
Hardswish
LeakyReLU
LogSigmoid
PReLU
ReLU
ReLU6
RReLU
SELU
CELU
GELU
Sigmoid
SiLU
Mish
Softplus
Softshrink
Softsign
Tanh
Tanhshrink
Threshold
GLU
Softmin
Softmax
Softmax2d
LogSoftmax
AdaptiveLogSoftmaxWithLoss
BatchNorm1d
BatchNorm2d
BatchNorm3d
LazyBatchNorm1d
LazyBatchNorm2d
LazyBatchNorm3d
GroupNorm
SyncBatchNorm
InstanceNorm1d
InstanceNorm2d
InstanceNorm3d
LazyInstanceNorm1d
LazyInstanceNorm2d
LazyInstanceNorm3d
LayerNorm
LocalResponseNorm
RNNBase
RNN
LSTM
GRU
RNNCell
LSTMCell
GRUCell
Transformer
TransformerEncoder
TransformerDecoder
TransformerEncoderLayer
TransformerDecoderLayer
Identity
Linear
Bilinear
LazyLinear
Dropout
Dropout1d
Dropout2d
Dropout3d
AlphaDropout
FeatureAlphaDropout
Embedding
EmbeddingBag
CosineSimilarity
PairwiseDistance
L1Loss
MSELoss
CrossEntropyLoss
CTCLoss
NLLLoss
PoissonNLLLoss
GaussianNLLLoss
KLDivLoss
BCELoss
BCEWithLogitsLoss
MarginRankingLoss
HingeEmbeddingLoss
MultiLabelMarginLoss
HuberLoss
SmoothL1Loss
SoftMarginLoss
MultiLabelSoftMarginLoss
CosineEmbeddingLoss
MultiMarginLoss
TripletMarginLoss
TripletMarginWithDistanceLoss
PixelShuffle
PixelUnshuffle
Upsample
UpsamplingNearest2d
UpsamplingBilinear2d
ChannelShuffle
DataParallel
parallel.DistributedDataParallel
Flatten
Unflatten
parameter.Parameter