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

时间:2020-08-05 10:32:58

标签: pytorch

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

1 个答案:

答案 0 :(得分:3)

.shape.size()的别名,并被添加为与numpy更紧密匹配,请参见此讨论here