PyTorch 2 · Tensors

Multiply matrices

Matrix multiplication with the @ operator.

y = a @ b

Official PyTorch 2 documentation

More tensors commands

All tensors commands · PyTorch 2 cheatsheet

Was this page helpful?