mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
Part of #123062 - #123062 Pull Request resolved: https://github.com/pytorch/pytorch/pull/128594 Approved by: https://github.com/mikaylagawarecki
335 lines
6.3 KiB
Python
335 lines
6.3 KiB
Python
from .module import Module # usort: skip
|
|
from .linear import Bilinear, Identity, LazyLinear, Linear # usort: skip
|
|
from .activation import (
|
|
CELU,
|
|
ELU,
|
|
GELU,
|
|
GLU,
|
|
Hardshrink,
|
|
Hardsigmoid,
|
|
Hardswish,
|
|
Hardtanh,
|
|
LeakyReLU,
|
|
LogSigmoid,
|
|
LogSoftmax,
|
|
Mish,
|
|
MultiheadAttention,
|
|
PReLU,
|
|
ReLU,
|
|
ReLU6,
|
|
RReLU,
|
|
SELU,
|
|
Sigmoid,
|
|
SiLU,
|
|
Softmax,
|
|
Softmax2d,
|
|
Softmin,
|
|
Softplus,
|
|
Softshrink,
|
|
Softsign,
|
|
Tanh,
|
|
Tanhshrink,
|
|
Threshold,
|
|
)
|
|
from .adaptive import AdaptiveLogSoftmaxWithLoss
|
|
from .batchnorm import (
|
|
BatchNorm1d,
|
|
BatchNorm2d,
|
|
BatchNorm3d,
|
|
LazyBatchNorm1d,
|
|
LazyBatchNorm2d,
|
|
LazyBatchNorm3d,
|
|
SyncBatchNorm,
|
|
)
|
|
from .channelshuffle import ChannelShuffle
|
|
from .container import (
|
|
Container,
|
|
ModuleDict,
|
|
ModuleList,
|
|
ParameterDict,
|
|
ParameterList,
|
|
Sequential,
|
|
)
|
|
from .conv import (
|
|
Conv1d,
|
|
Conv2d,
|
|
Conv3d,
|
|
ConvTranspose1d,
|
|
ConvTranspose2d,
|
|
ConvTranspose3d,
|
|
LazyConv1d,
|
|
LazyConv2d,
|
|
LazyConv3d,
|
|
LazyConvTranspose1d,
|
|
LazyConvTranspose2d,
|
|
LazyConvTranspose3d,
|
|
)
|
|
from .distance import CosineSimilarity, PairwiseDistance
|
|
from .dropout import (
|
|
AlphaDropout,
|
|
Dropout,
|
|
Dropout1d,
|
|
Dropout2d,
|
|
Dropout3d,
|
|
FeatureAlphaDropout,
|
|
)
|
|
from .flatten import Flatten, Unflatten
|
|
from .fold import Fold, Unfold
|
|
from .instancenorm import (
|
|
InstanceNorm1d,
|
|
InstanceNorm2d,
|
|
InstanceNorm3d,
|
|
LazyInstanceNorm1d,
|
|
LazyInstanceNorm2d,
|
|
LazyInstanceNorm3d,
|
|
)
|
|
from .loss import (
|
|
BCELoss,
|
|
BCEWithLogitsLoss,
|
|
CosineEmbeddingLoss,
|
|
CrossEntropyLoss,
|
|
CTCLoss,
|
|
GaussianNLLLoss,
|
|
HingeEmbeddingLoss,
|
|
HuberLoss,
|
|
KLDivLoss,
|
|
L1Loss,
|
|
MarginRankingLoss,
|
|
MSELoss,
|
|
MultiLabelMarginLoss,
|
|
MultiLabelSoftMarginLoss,
|
|
MultiMarginLoss,
|
|
NLLLoss,
|
|
NLLLoss2d,
|
|
PoissonNLLLoss,
|
|
SmoothL1Loss,
|
|
SoftMarginLoss,
|
|
TripletMarginLoss,
|
|
TripletMarginWithDistanceLoss,
|
|
)
|
|
from .normalization import (
|
|
CrossMapLRN2d,
|
|
GroupNorm,
|
|
LayerNorm,
|
|
LocalResponseNorm,
|
|
RMSNorm,
|
|
)
|
|
from .padding import (
|
|
CircularPad1d,
|
|
CircularPad2d,
|
|
CircularPad3d,
|
|
ConstantPad1d,
|
|
ConstantPad2d,
|
|
ConstantPad3d,
|
|
ReflectionPad1d,
|
|
ReflectionPad2d,
|
|
ReflectionPad3d,
|
|
ReplicationPad1d,
|
|
ReplicationPad2d,
|
|
ReplicationPad3d,
|
|
ZeroPad1d,
|
|
ZeroPad2d,
|
|
ZeroPad3d,
|
|
)
|
|
from .pixelshuffle import PixelShuffle, PixelUnshuffle
|
|
from .pooling import (
|
|
AdaptiveAvgPool1d,
|
|
AdaptiveAvgPool2d,
|
|
AdaptiveAvgPool3d,
|
|
AdaptiveMaxPool1d,
|
|
AdaptiveMaxPool2d,
|
|
AdaptiveMaxPool3d,
|
|
AvgPool1d,
|
|
AvgPool2d,
|
|
AvgPool3d,
|
|
FractionalMaxPool2d,
|
|
FractionalMaxPool3d,
|
|
LPPool1d,
|
|
LPPool2d,
|
|
LPPool3d,
|
|
MaxPool1d,
|
|
MaxPool2d,
|
|
MaxPool3d,
|
|
MaxUnpool1d,
|
|
MaxUnpool2d,
|
|
MaxUnpool3d,
|
|
)
|
|
from .rnn import GRU, GRUCell, LSTM, LSTMCell, RNN, RNNBase, RNNCell, RNNCellBase
|
|
from .sparse import Embedding, EmbeddingBag
|
|
from .transformer import (
|
|
Transformer,
|
|
TransformerDecoder,
|
|
TransformerDecoderLayer,
|
|
TransformerEncoder,
|
|
TransformerEncoderLayer,
|
|
)
|
|
from .upsampling import Upsample, UpsamplingBilinear2d, UpsamplingNearest2d
|
|
|
|
|
|
__all__ = [
|
|
"AdaptiveAvgPool1d",
|
|
"AdaptiveAvgPool2d",
|
|
"AdaptiveAvgPool3d",
|
|
"AdaptiveLogSoftmaxWithLoss",
|
|
"AdaptiveMaxPool1d",
|
|
"AdaptiveMaxPool2d",
|
|
"AdaptiveMaxPool3d",
|
|
"AlphaDropout",
|
|
"AvgPool1d",
|
|
"AvgPool2d",
|
|
"AvgPool3d",
|
|
"BCELoss",
|
|
"BCEWithLogitsLoss",
|
|
"BatchNorm1d",
|
|
"BatchNorm2d",
|
|
"BatchNorm3d",
|
|
"Bilinear",
|
|
"CELU",
|
|
"CTCLoss",
|
|
"ChannelShuffle",
|
|
"CircularPad1d",
|
|
"CircularPad2d",
|
|
"CircularPad3d",
|
|
"ConstantPad1d",
|
|
"ConstantPad2d",
|
|
"ConstantPad3d",
|
|
"Container",
|
|
"Conv1d",
|
|
"Conv2d",
|
|
"Conv3d",
|
|
"ConvTranspose1d",
|
|
"ConvTranspose2d",
|
|
"ConvTranspose3d",
|
|
"CosineEmbeddingLoss",
|
|
"CosineSimilarity",
|
|
"CrossEntropyLoss",
|
|
"CrossMapLRN2d",
|
|
"Dropout",
|
|
"Dropout1d",
|
|
"Dropout2d",
|
|
"Dropout3d",
|
|
"ELU",
|
|
"Embedding",
|
|
"EmbeddingBag",
|
|
"FeatureAlphaDropout",
|
|
"Flatten",
|
|
"Fold",
|
|
"FractionalMaxPool2d",
|
|
"FractionalMaxPool3d",
|
|
"GELU",
|
|
"GLU",
|
|
"GRU",
|
|
"GRUCell",
|
|
"GaussianNLLLoss",
|
|
"GroupNorm",
|
|
"Hardshrink",
|
|
"Hardsigmoid",
|
|
"Hardswish",
|
|
"Hardtanh",
|
|
"HingeEmbeddingLoss",
|
|
"HuberLoss",
|
|
"Identity",
|
|
"InstanceNorm1d",
|
|
"InstanceNorm2d",
|
|
"InstanceNorm3d",
|
|
"KLDivLoss",
|
|
"L1Loss",
|
|
"LPPool1d",
|
|
"LPPool2d",
|
|
"LPPool3d",
|
|
"LSTM",
|
|
"LSTMCell",
|
|
"LayerNorm",
|
|
"LazyBatchNorm1d",
|
|
"LazyBatchNorm2d",
|
|
"LazyBatchNorm3d",
|
|
"LazyConv1d",
|
|
"LazyConv2d",
|
|
"LazyConv3d",
|
|
"LazyConvTranspose1d",
|
|
"LazyConvTranspose2d",
|
|
"LazyConvTranspose3d",
|
|
"LazyInstanceNorm1d",
|
|
"LazyInstanceNorm2d",
|
|
"LazyInstanceNorm3d",
|
|
"LazyLinear",
|
|
"LeakyReLU",
|
|
"Linear",
|
|
"LocalResponseNorm",
|
|
"LogSigmoid",
|
|
"LogSoftmax",
|
|
"MSELoss",
|
|
"MarginRankingLoss",
|
|
"MaxPool1d",
|
|
"MaxPool2d",
|
|
"MaxPool3d",
|
|
"MaxUnpool1d",
|
|
"MaxUnpool2d",
|
|
"MaxUnpool3d",
|
|
"Mish",
|
|
"Module",
|
|
"ModuleDict",
|
|
"ModuleList",
|
|
"MultiLabelMarginLoss",
|
|
"MultiLabelSoftMarginLoss",
|
|
"MultiMarginLoss",
|
|
"MultiheadAttention",
|
|
"NLLLoss",
|
|
"NLLLoss2d",
|
|
"PReLU",
|
|
"PairwiseDistance",
|
|
"ParameterDict",
|
|
"ParameterList",
|
|
"PixelShuffle",
|
|
"PixelUnshuffle",
|
|
"PoissonNLLLoss",
|
|
"RMSNorm",
|
|
"RNN",
|
|
"RNNBase",
|
|
"RNNCell",
|
|
"RNNCellBase",
|
|
"RReLU",
|
|
"ReLU",
|
|
"ReLU6",
|
|
"ReflectionPad1d",
|
|
"ReflectionPad2d",
|
|
"ReflectionPad3d",
|
|
"ReplicationPad1d",
|
|
"ReplicationPad2d",
|
|
"ReplicationPad3d",
|
|
"SELU",
|
|
"Sequential",
|
|
"SiLU",
|
|
"Sigmoid",
|
|
"SmoothL1Loss",
|
|
"SoftMarginLoss",
|
|
"Softmax",
|
|
"Softmax2d",
|
|
"Softmin",
|
|
"Softplus",
|
|
"Softshrink",
|
|
"Softsign",
|
|
"SyncBatchNorm",
|
|
"Tanh",
|
|
"Tanhshrink",
|
|
"Threshold",
|
|
"Transformer",
|
|
"TransformerDecoder",
|
|
"TransformerDecoderLayer",
|
|
"TransformerEncoder",
|
|
"TransformerEncoderLayer",
|
|
"TripletMarginLoss",
|
|
"TripletMarginWithDistanceLoss",
|
|
"Unflatten",
|
|
"Unfold",
|
|
"Upsample",
|
|
"UpsamplingBilinear2d",
|
|
"UpsamplingNearest2d",
|
|
"ZeroPad1d",
|
|
"ZeroPad2d",
|
|
"ZeroPad3d",
|
|
]
|
|
|
|
# Please keep this list sorted
|
|
assert __all__ == sorted(__all__)
|