This module contains various PyTorch modules and related APIs to help with building new model architectures. It follows similar conventions to torch.nn and can be considered an addendum to it.
torch.nn