gpt4 book ai didi

python - 如何将张量逐行乘以 PyTorch 中的向量?

转载 作者:行者123 更新时间:2023-12-04 23:36:15 24 4
gpt4 key购买 nike

当我有张量时 m形状[12, 10]和一个向量 s具有形状的标量 [12] ,如何乘以每一行 ms 中的相应标量?

最佳答案

您需要添加相应的单例维度:

m * s[:, None]
s[:, None]大小为 (12, 1)当乘以 (12, 10)张量由 (12, 1)张量 pytoch 知道 broadcast s沿着第二个单例维度并正确执行“逐元素”乘积。

关于python - 如何将张量逐行乘以 PyTorch 中的向量?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/53987906/

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