如何在pytorch中对两个4D不等大小张量进行元素乘法运算？

Question

我有一个张量A和Tensor B.

大小A = [2,64,56,56]

B的大小= [2,64,29,29]

我怎么能执行torch.mul(A,B)？张量的大小不等。

RuntimeError：shape [2, 64, 56, 56]对于大小为107648的输入无效

Answer 1

你可以在这里查看文档：https://pytorch.org/docs/stable/torch.html#torch.mul

在那里，你可以阅读：

输入和其他形状必须是可播放的。

你可以在这里阅读有关广播性的信息：https://pytorch.org/docs/stable/notes/broadcasting.html#broadcasting-semantics

最后，考虑一下你想要达到的目标可能是有意义的。我不确定你想要做什么这样的操作，因为要有元素乘法，你需要张量具有相同的形状。