PyTorch: nn — PyTorch Tutorials 1.7.0 documentation
pytorch.org › tutorials › beginnerA fully-connected ReLU network with one hidden layer, trained to predict y from x by minimizing squared Euclidean distance. This implementation uses the nn package from PyTorch to build the network. PyTorch autograd makes it easy to define computational graphs and take gradients, but raw autograd can be a bit too low-level for defining complex ...