pytorch中两个4D大小不等的张量如何进行元素明智乘法？

Question

我有张量A和张量B。

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

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

如何执行torch.mul(A,B)？张量大小不相等。

  

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

Answer 1

您可以在此处查看文档：{​​{3}}

在那里，您可以阅读：

  

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

您可以在此处阅读有关广播性的信息：https://pytorch.org/docs/stable/torch.html#torch.mul

最后，考虑要实现的目标可能很有意义。我不确定您要进行这种操作的原因，因为要进行逐元素乘法，就需要张量具有相同的形状。