torch.matmul

torch.matmul 是 PyTorch 中用于矩阵乘法的函数。它可以对两个张量进行矩阵乘法操作，并返回结果张量。例如：

import torch

# 创建两个张量
a = torch.tensor([[1, 2], [3, 4]])
b = torch.tensor([[5, 6], [7, 8]])

# 矩阵乘法
c = torch.matmul(a, b)

print(c)

输出结果为：

tensor([[19, 22],
        [43, 50]])

其中，结果张量 c 的第一个元素为 1*5+2*7=19，第二个元素为 1*6+2*8=22，以此类推。