PyTorch: Difference between revisions

Revision as of 17:06, 17 February 2024

Revision as of 03:43, 19 February 2024

Line 58:

{{main|Tensor (machine learning)}}

PyTorch defines a class 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 ==