docs/source/nn.rst

torch.nn
===================================

.. automodule:: torch.nn
.. currentmodule:: torch.nn

Containers
----------------------------------

.. autoclass:: Module
    :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:

Multi-GPU layers
----------------
.. autoclass:: DataParallel
    :members: