torch.mm vs torch.matmul

황준하·2024년 3월 24일

torch.mm

torch.mm(input, mat2, *, out=None) → Tensor

torch.mm은 Broadcast를 허용하지 않음

inputmat2간의 행렬곱을 수행하며, 2차원 행렬끼리의 곱만 지원한다.

torch.matmul

torch.matmul(input, other, *, out=None) → Tensor

torch.matmul은 Broadcast를 지원함

0개의 댓글