gpt4 book ai didi

python - 乘以 PyTorch 张量的所有元素

转载 作者:行者123 更新时间:2023-12-03 15:36:07 26 4
gpt4 key购买 nike

我想做这样的事情 question在 PyTorch 中,即将所有元素相互乘以保持某个轴恒定。 PyTorch 是否有任何预定义的功能?

最佳答案

是的。 torch.prod .使用 dim参数来告诉您希望沿着哪个轴计算产品。

x = torch.randn((2, 2))
print(x)
print(torch.prod(x, 0)) # product along 0th axis

这打印
tensor([[-0.3661, 1.0693],
[0.5144, 1.3489]])
tensor([-0.1883, 1.4424])

关于python - 乘以 PyTorch 张量的所有元素,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/56592745/

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