Revision as of 21:52, 1 April 2024 edit Dujo (talk \| contribs) Extended confirmed users 1,187 edits →PyTorch tensors: Add reference ← Previous edit		Revision as of 09:53, 7 April 2024 edit undo 134.2.251.2 (talk) Change "class" to "module". The nn module of PyTorch is not a class. Next edit →
Line 46: {{main\|Tensor (machine learning)}} PyTorch defines a ~~class~~module called nn (<code>torch.nn</code>) to describe neural networks and to support training. This module offers a comprehensive collection of building blocks for neural networks, including various layers and activation functions, enabling the construction of complex models. == Example ==

PyTorch: Difference between revisions