torch.mm(input, mat2, *, out=None) → Tensor
torch.mm은 Broadcast를 허용하지 않음
torch.mm
input과 mat2간의 행렬곱을 수행하며, 2차원 행렬끼리의 곱만 지원한다.
input
mat2
torch.matmul(input, other, *, out=None) → Tensor
torch.matmul은 Broadcast를 지원함
torch.matmul