torch.nn — PyTorch 2.5 documentation
pytorch.org