Tensor.shape

property Tensor.shape

Number of elements per axis.