Was this section helpful?
torch.nn
- PyTorch Documentation, PyTorch Contributors, 2024 (PyTorch Foundation) - The official documentation for PyTorch's neural network module, which is the base class for all network modules. It details the methods and attributes relevant for architecture and parameter inspection.state_dict
which is also a key tool for parameter access and inspection.