torch.nn.modules.module — PyTorch 2.5 documentation

pytorch.org pytorch.org