Pytorch中的Tensor.size和Tensor.shape有什么不同?我想知道张量的元素数和维数。例如,对于维度为2×3×4的张量,我预计元素数量为24,维度为(2,3,4)。谢谢。
Tensor.size
Tensor.shape
发布于 2020-08-05 18:36:42
.shape是.size()的别名,添加它是为了更好地匹配numpy,请参阅此讨论here。
.shape
.size()
numpy
https://stackoverflow.com/questions/63263292
相似问题