doc torch.mm code example
Example: pytorch - matrix multiplication
# Matrix multiplication
a = torch.tensor([[0, 2, 4], [1, 3, 5]])
b = torch.tensor([[0, 7], [8, 9], [10, 11]])
torch.mm(a, b)
# Matrix multiplication
a = torch.tensor([[0, 2, 4], [1, 3, 5]])
b = torch.tensor([[0, 7], [8, 9], [10, 11]])
torch.mm(a, b)