torch.Tensor.size — PyTorch 2.7 documentation (original) (raw)
- Docs >
- torch.Tensor >
- torch.Tensor.size
Shortcuts
Tensor.size(dim=None) → torch.Size or int¶
Returns the size of the self tensor. If dim is not specified, the returned value is a torch.Size, a subclass of tuple. If dim is specified, returns an int holding the size of that dimension.
Parameters
dim (int, optional) – The dimension for which to retrieve the size.
Example:
t = torch.empty(3, 4, 5) t.size() torch.Size([3, 4, 5]) t.size(dim=1) 4