In [1]:
import torch
In [2]:
mat_a = torch.tensor([0, 3, 5, 5, 5, 2]).view(2, 3)
mat_a
Out[2]:
tensor([[0, 3, 5],
        [5, 5, 2]])
In [3]:
mat_b = torch.tensor([3, 4, 3, -2, 4, -2]).view(3, 2)
mat_b
Out[3]:
tensor([[ 3,  4],
        [ 3, -2],
        [ 4, -2]])
In [4]:
torch.matmul(mat_a, mat_b)
Out[4]:
tensor([[ 29, -16],
        [ 38,   6]])
In [5]:
mat_a @ mat_b
Out[5]:
tensor([[ 29, -16],
        [ 38,   6]])