torch.baddbmm

明天日和会不会去航海

于 2022-03-21 20:25:10 发布

阅读量6.2k

点赞数 1

文章标签： pytorch

本文链接：https://blog.csdn.net/qq_41289353/article/details/123644925

版权

torch.baddbmm函数用于执行批量矩阵乘法并将其结果加到输入张量上。该操作涉及三个张量：input、batch1和batch2，它们的尺寸需满足特定条件。例如，如果batch1的尺寸为(b*n*m)，batch2的尺寸为(b*m*p)，则input必须是(b*n*p)的尺寸。通过这个函数，可以将batch1和batch2的批量矩阵乘积累加到input上，最终输出的尺寸保持不变。在给定的例子中，创建了三个张量并应用torch.baddbmm，输出的尺寸为torch.Size([10,3,5])。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

torch.baddbmm(input, batch1, batch2, *, beta=1, alpha=1, out=None) → Tensor

batch1和batch2执行批量的矩阵相乘，然后和input相加，得到output

如果batch1 是（b*n*m）的tensor，batch2是（b*m*p）的tensor，那么input必须是（b*n*p）的tensor.

output = input + batch1 * batch2

参数

输入( Tensor ) – 要添加的张量

batch1 ( Tensor ) – 要相乘的第一批矩阵

batch2 ( Tensor ) – 第二批要相乘的矩阵

M = torch.randn(10, 3, 5)
batch1 = torch.randn(10, 3, 4)
batch2 = torch.randn(10, 4, 5)
torch.baddbmm(M, batch1, batch2).size()

输出：

torch.Size([10, 3, 5])