pytorch 批矩阵相乘

在PyTorch中，可以使用torch.matmul函数来进行批矩阵相乘。批矩阵相乘是指同时对多个矩阵进行相乘操作。

假设我们有两个批次的矩阵A和B，它们的形状分别为(batch_size, n, m)和(batch_size, m, p)，其中batch_size表示批次大小，n、m、p分别表示矩阵的行数和列数。要计算这两个批次矩阵的相乘结果，可以使用torch.matmul函数进行如下操作：

import torch

# 定义两个批次的矩阵A和B
A = torch.randn(2, 3, 4)
B = torch.randn(2, 4, 5)

# 批矩阵相乘
C = torch.matmul(A, B)

print(C.shape)  # 输出结果的形状为(batch_size, n, p)

在上述代码中，我们首先使用torch.randn函数生成了两个批次的随机矩阵A和B，然后使用torch.matmul函数对它们进行相乘操作，得到结果矩阵C。最后打印出结果矩阵C的形状。