标签: pytorch
Pytorch中的Tensor.size和Tensor.shape有什么区别? 我想获取张量的元素数量和尺寸。例如,对于张量为2 x 3 x 4的张量,我期望元素数量为24,尺寸为(2,3,4)。 谢谢。
Tensor.size
Tensor.shape
答案 0 :(得分:3)
.shape是.size()的别名,并被添加为与numpy更紧密匹配,请参见此讨论here。
.shape
.size()
numpy