pytorch torch.bmm

`torch.bmm`是PyTorch中的一个函数，用于计算两个tensor的批矩阵乘法。其中，第一个tensor的形状为(*, n, m)，第二个tensor的形状为(*, m, p)，结果tensor的形状为(*, n, p)。具体来说，对于第一个tensor的第i个矩阵和第二个tensor的第i个矩阵，进行矩阵乘法操作并返回结果tensor中的第i个矩阵。其中，*表示任意的额外维度。

示例代码如下：

import torch

# 创建两个tensor
a = torch.randn(10, 3, 4)
b = torch.randn(10, 4, 5)

# 进行批矩阵乘法操作
c = torch.bmm(a, b)

# 输出结果tensor的形状
print(c.shape)  # 输出：torch.Size([10, 3, 5])

在这个例子中，a的形状为(10, 3, 4)，b的形状为(10, 4, 5)，因此结果tensor的形状为(10, 3, 5)。