When we define a Tensor, the size()
is equivalent to .shape
, i.e.
a = torch.Tensor([1, 2, 3])
assert a.size() == a.shape
I could not find the documentation about .shape
When we define a Tensor, the size()
is equivalent to .shape
, i.e.
a = torch.Tensor([1, 2, 3])
assert a.size() == a.shape
I could not find the documentation about .shape