[人工智能]pytoch矩阵乘法torch.bmm

????????torch.bmm函数可实现矩阵乘法。pytorch官方文档地址：torch.bmm — PyTorch 1.10.1 documentationhttps://pytorch.org/docs/stable/generated/torch.bmm.html#torch.bmm

函数定义：

torch.bmm(input, mat2, *, deterministic=False, out=None) → Tensor

功能：

????????对 input 和 mat2 矩阵执行批处理矩阵积。 input 和 mat2 必须是三维张量，每个张量包含相同数量的矩阵。

输入：

input tensor 维度：(b×n×m);
mat2 tensor 维度：(b×m×p);

输出：

out tensor 维度： (b×n×p)

????????对于矩阵运算AxB，假设A为n*m维，B维m*p维，则结果为n*p维。b为batch大小。

测试：

input = torch.randn(10, 3, 4) 
mat2 = torch.randn(10, 4, 5) 
res = torch.bmm(input, mat2) 
print(res.size()) 
=> 
torch.Size([10, 3, 5])