

Serialization


Parallelism


Locally disabling gradient computation


Math operations


Utilities


Symbolic Numbers


Optimizations


Operator Tags


Engine Configuration


torch.nn


Parameter


UninitializedParameter


UninitializedBuffer


Containers


Convolution Layers


Pooling layers


Padding Layers


Non-linear Activations (weighted sum, nonlinearity)


Non-linear Activations (other)


Normalization Layers


Recurrent Layers


Transformer Layers


Linear Layers


Dropout Layers


Sparse Layers


Distance Functions


Loss Functions


Vision Layers


Shuffle Layers


DataParallel Layers (multi-GPU, distributed)


Utilities


Quantized Functions


Lazy Modules Initialization


torch.nn.functional


Convolution functions


Pooling functions


Non-linear activation functions


Linear functions


Dropout functions


Sparse functions


Distance functions


Loss functions

