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.