Skip to content

Glossary: Linear Algebra for ML

PyTorch

A Python library for building and training neural networks. Its `nn.Linear` layer stores its weight matrix as (outputs × inputs) and computes `X @ W.T + b`, the transpose of the (inputs × outputs) layout used with `X @ W`.

Where the book teaches it

Related terms

Neural networkNumPy