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

Question

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

Answer 1

.shape是 的别名.size()，并被添加为更紧密匹配numpy，请参阅此处的讨论。

Answer 2

.shape 是张量的属性，而 size() 是一个函数。它们都返回相同的值。