core/nn/module library

Base class for stateful neural-network modules.

Mirrors PyTorch's nn.Module: a module bundles trainable parameters and defines a forward pass via call. Subclasses expose their trainable tensors via parameters, which optimizers iterate.

Classes

Module