mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
239 lines
3.2 KiB
ReStructuredText
239 lines
3.2 KiB
ReStructuredText
torch.nn
|
|
===================================
|
|
|
|
.. automodule:: torch.nn
|
|
.. currentmodule:: torch.nn
|
|
|
|
Containers
|
|
----------------------------------
|
|
|
|
.. autoclass:: Module
|
|
:members:
|
|
.. autoclass:: Container
|
|
:members:
|
|
|
|
Convolution Layers
|
|
----------------------------------
|
|
|
|
.. autoclass:: Conv1d
|
|
:members:
|
|
|
|
.. autoclass:: Conv2d
|
|
:members:
|
|
|
|
.. autoclass:: Conv3d
|
|
:members:
|
|
|
|
.. autoclass:: ConvTranspose1d
|
|
:members:
|
|
|
|
.. autoclass:: ConvTranspose2d
|
|
:members:
|
|
|
|
.. autoclass:: ConvTranspose3d
|
|
:members:
|
|
|
|
|
|
Pooling Layers
|
|
----------------------------------
|
|
|
|
.. autoclass:: MaxPool1d
|
|
:members:
|
|
|
|
.. autoclass:: MaxPool2d
|
|
:members:
|
|
|
|
.. autoclass:: MaxPool3d
|
|
:members:
|
|
|
|
.. autoclass:: MaxUnpool1d
|
|
:members:
|
|
|
|
.. autoclass:: MaxUnpool2d
|
|
:members:
|
|
|
|
.. autoclass:: MaxUnpool3d
|
|
:members:
|
|
|
|
.. autoclass:: AvgPool1d
|
|
:members:
|
|
|
|
.. autoclass:: AvgPool2d
|
|
:members:
|
|
|
|
.. autoclass:: AvgPool3d
|
|
:members:
|
|
|
|
.. autoclass:: FractionalMaxPool2d
|
|
:members:
|
|
|
|
.. autoclass:: LPPool2d
|
|
:members:
|
|
|
|
Non-linear Activations
|
|
----------------------------------
|
|
|
|
.. autoclass:: ReLU
|
|
:members:
|
|
|
|
.. autoclass:: ReLU6
|
|
:members:
|
|
|
|
.. autoclass:: ELU
|
|
:members:
|
|
|
|
.. autoclass:: PReLU
|
|
:members:
|
|
|
|
.. autoclass:: LeakyReLU
|
|
:members:
|
|
|
|
.. autoclass:: Threshold
|
|
:members:
|
|
|
|
.. autoclass:: Hardtanh
|
|
:members:
|
|
|
|
.. autoclass:: Sigmoid
|
|
:members:
|
|
|
|
.. autoclass:: Tanh
|
|
:members:
|
|
|
|
.. autoclass:: LogSigmoid
|
|
:members:
|
|
|
|
.. autoclass:: Softplus
|
|
:members:
|
|
|
|
.. autoclass:: Softshrink
|
|
:members:
|
|
|
|
.. autoclass:: Softsign
|
|
:members:
|
|
|
|
.. autoclass:: Tanhshrink
|
|
:members:
|
|
|
|
.. autoclass:: Softmin
|
|
:members:
|
|
|
|
.. autoclass:: Softmax
|
|
:members:
|
|
|
|
.. autoclass:: LogSoftmax
|
|
:members:
|
|
|
|
|
|
Normalization layers
|
|
----------------------------------
|
|
|
|
.. autoclass:: BatchNorm1d
|
|
:members:
|
|
|
|
.. autoclass:: BatchNorm2d
|
|
:members:
|
|
|
|
.. autoclass:: BatchNorm3d
|
|
:members:
|
|
|
|
|
|
Recurrent layers
|
|
----------------------------------
|
|
|
|
.. autoclass:: RNN
|
|
:members:
|
|
|
|
.. autoclass:: LSTM
|
|
:members:
|
|
|
|
.. autoclass:: GRU
|
|
:members:
|
|
|
|
.. autoclass:: RNNCell
|
|
:members:
|
|
|
|
.. autoclass:: LSTMCell
|
|
:members:
|
|
|
|
.. autoclass:: GRUCell
|
|
:members:
|
|
|
|
Linear layers
|
|
----------------------------------
|
|
|
|
.. autoclass:: Linear
|
|
:members:
|
|
|
|
|
|
Dropout layers
|
|
----------------------------------
|
|
|
|
.. autoclass:: Dropout
|
|
:members:
|
|
|
|
.. autoclass:: Dropout2d
|
|
:members:
|
|
|
|
.. autoclass:: Dropout3d
|
|
:members:
|
|
|
|
|
|
Sparse layers
|
|
----------------------------------
|
|
|
|
.. autoclass:: Embedding
|
|
:members:
|
|
|
|
|
|
Loss functions
|
|
----------------------------------
|
|
|
|
.. autoclass:: L1Loss
|
|
:members:
|
|
|
|
.. autoclass:: MSELoss
|
|
:members:
|
|
|
|
.. autoclass:: CrossEntropyLoss
|
|
:members:
|
|
|
|
.. autoclass:: NLLLoss
|
|
:members:
|
|
|
|
.. autoclass:: KLDivLoss
|
|
:members:
|
|
|
|
.. autoclass:: BCELoss
|
|
:members:
|
|
|
|
.. autoclass:: MarginRankingLoss
|
|
:members:
|
|
|
|
.. autoclass:: HingeEmbeddingLoss
|
|
:members:
|
|
|
|
.. autoclass:: MultiLabelMarginLoss
|
|
:members:
|
|
|
|
.. autoclass:: SmoothL1Loss
|
|
:members:
|
|
|
|
.. autoclass:: SoftMarginLoss
|
|
:members:
|
|
|
|
.. autoclass:: MultiLabelSoftMarginLoss
|
|
:members:
|
|
|
|
.. autoclass:: CosineEmbeddingLoss
|
|
:members:
|
|
|
|
.. autoclass:: MultiMarginLoss
|
|
:members:
|
|
|
|
|
|
Vision layers
|
|
----------------
|
|
.. autoclass:: PixelShuffle
|
|
:members:
|