torch.nn — PyTorch 2.10 documentation
pytorch.org