torch.matmul(): Înmulțirea Matricială a Tensorilor în PyTorch & Deep Learning Practic

Funcția torch.matmul() (accesibilă și prin operatorul infix @) efectuează produsul matricial între doi tensori, adaptându-se automat la numărul de dimensiuni: produs scalar pentru vectori 1D, înmulțire clasică 2D sau înmulțire pe loturi dacă tensorii au 3 sau mai multe axe.

Exemplu în cod

import torch

A = torch.randn(2, 3)
B = torch.randn(3, 4)

# Înmulțire matricială standard
C = torch.matmul(A, B)
# Sau folosind operatorul @
C_alias = A @ B

print("Forma rezultat:", C.shape)  # torch.Size([2, 4])

Este operația de bază din straturile dense Linear și mecanismul de atenție $Q \cdot K^T$ din Transformere.