gpt4 book ai didi

pytorch - PyTorch 中的 Tensor.size 和 Tensor.shape 有什么区别?

转载 作者:行者123 更新时间:2023-12-03 20:04:43 28 4
gpt4 key购买 nike

Pytorch 中的 Tensor.sizeTensor.shape 有什么区别?
我想获取元素的数量和张量的维度。例如,对于尺寸为 2 x 3 x 4 的张量,我希望元素数量为 24,维度为 (2,3,4)。
谢谢。

最佳答案

.shape.size() 的别名,并且被添加以更接近匹配 numpy ,请参阅此讨论 here

关于pytorch - PyTorch 中的 Tensor.size 和 Tensor.shape 有什么区别?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/63263292/

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