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

Question

25

在Pytorch中，Tensor.size和Tensor.shape有什么区别呢？我想得到张量的元素数量和维度。例如，对于一个具有2×3×4维度的张量，我期望得到元素数为24，维度为(2,3,4)。谢谢。

- Peyman habibi

谢谢。不，它不一样。 - Peyman habibi

对于元素数量，我使用了 torch.numel(var)。 - Peyman habibi

2个回答

4

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

- former_Epsilon

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Tasnuva Leeya · Accepted Answer

.shape 是 .size() 的别名，它被添加的目的是更加贴近 numpy。具体讨论请参考此处。