我有两个 PyTorch 张量 A，形状为 [30, 11, 32, 64] 和 B，形状为 [30, 11, 89]。 我的目标是创建一个形状为 [30, 11, 89, 32, 64] 的张量 C，其中张量 A 应该在第三维中扩展和复制，然后在适当的维度与张量 B 相乘。
如何在 PyTorch 中实现这一点？
最佳答案
您可以使用与 numpy 相同的广播语义。实现此目的的一种方法是
C = A.reshape(30, 11, 1, 32, 64) * B.reshape(30, 11, 89, 1, 1)
关于python - Pytorch 乘法广播，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/59732266/