问Pytorch乘法广播
EN

Stack Overflow用户

提问于 2020-01-14 18:48:47

回答 1查看 166关注 0票数 1

我有两个PyTorch张量A，形状30，11，32，64和B，形状30，11，89。我的目标是创建一个形状为30，11，89，32，64的张量C，其中张量A应该扩展并在第三维复制，然后在适当的维度与张量B相乘。

如何在PyTorch中实现这一点？

发布于 2020-01-14 23:25:39

您可以使用与numpy相同的广播语义。实现这一目标的一种方法是

C = A.reshape(30, 11, 1, 32, 64) * B.reshape(30, 11, 89, 1, 1)

票数 0

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/59732266

复制

相似问题

问Pytorch乘法广播EN