gpt4 book ai didi

python - Pytorch 乘法广播

转载 作者:太空宇宙 更新时间:2023-11-03 19:52:41 24 4
gpt4 key购买 nike

我有两个 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/

24 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com