'What is the difference between Tensor.size and Tensor.shape in PyTorch?
What is the difference between Tensor.size and Tensor.shape in Pytorch?
I want to get the number of elements and the dimensions of Tensor. For example for a tensor with the dimensions of 2 by 3 by 4 I expect 24 for number of elements and (2,3,4) for dimension.
Thanks.
Solution 1:[1]
.shape is an alias for .size(), and was added to more closely match numpy, see this discussion here.
Solution 2:[2]
.shape is an attribute of the tensor whereas size() is a function. They both return the same value.
Sources
This article follows the attribution requirements of Stack Overflow and is licensed under CC BY-SA 3.0.
Source: Stack Overflow
| Solution | Source |
|---|---|
| Solution 1 | Mughees |
| Solution 2 | former_Epsilon |