Content deleted Content added
BegbertBiggs (talk | contribs) →PyTorch neural networks: more fitting main article rather than duplicate |
No edit summary |
||
Line 46:
{{main|Neural network (machine learning)}}
PyTorch defines a 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. Custom networks are built by inheriting from the <code>torch.nn</code> module and defining the <code>forwaed()</code> function.
== Example ==
| |||