pytorch：如何将三维张量与二维张量相乘

我得到了一个三维张量

三个

和一个二维张量

两个

，它们需要相乘。例如，尺寸为：

three.shape = 4x100x700
two.shape = 4x100

输出形状应为：

output.shape = 4x100x700

---

所以基本上，在

输出[a，b]

中应该有700个标量，这些标量是通过将

三个[a，b]

中的所有700个标量与

两个[a，b]

中的单个标量相乘来计算的，您可以简单地向

两个[a，b]
添加一个额外的维度：

output = three * two.unsqueeze(-1)

还有其他语法，例如：

output = three * two[..., None]