oneflow.Tensor.size¶
-
Tensor.
size
()¶ Returns the size of the self tensor. If dim is not specified, the returned value is a oneflow.Size, a subclass of tuple. If dim is specified, returns an int holding the size of that dimension.
The interface is consistent with PyTorch.
- Parameters
idx (int, optional) – The dimension for which to retrieve the size.