pytorch torch.matmul函数介绍
torch.matmul 是 PyTorch 中用于进行矩阵乘法的函数。它可以执行两维矩阵、向量和更高维张量之间的乘法运算，支持的运算取决于输入张量的维度。
1. 函数签名
torch.matmul(input, other, out=None)
- input: 左乘的张量。
- other: 右乘的张量。
- out: 可选，用于存储输出结果的张量。
2. 不同维度的乘法规则
torch.matmul 根据输入张量的维度执行不同类型的乘法：
- 两个标量（0维）：相当于两个数的普通乘法。
- 一个向量（1维）与一个矩阵（2维）：
- 若第一个张量是一维向量，第二个是二维矩阵，则执行向量和矩阵的乘法，输出是一个向量。
- 若第一个张量是二维矩阵，第二个是一维向量，则执行矩阵和向量的乘法，输出是一个向量。
- 两个矩阵（2维）：执行标准的矩阵乘法（即行×列相乘）。
- 高维张量：高维张量可以被看作是包含多个矩阵，
torch.matmul将沿着批量维度（batch dimension）执行批量矩阵乘法。