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)