小编Ami*_*ibi的帖子

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

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

pytorch

6
推荐指数
2
解决办法
3380
查看次数

标签 统计

pytorch ×1