torch.nn [1]

Containers

Module

Base class for all neural network modules

Sequential

A sequential container

Module List

Holds submodules in a list

Module Dict

Holds submodules in a dictionary

ParameterList

Holds parameters in a list

ParameterDict

Holds parameters in a dictionary

Reference

1. https://pytorch.org/docs/stable/nn.html#containers