Tensor.shape

Returns the shape of the tensor.

Usage

from tinygrad.tensor import Tensor
 
# create a tensor
tensor = Tensor([[1, 2, 3, 4, 5], [6, 7, 8, 9, 10]])
 
# get the shape
print(tensor.shape)

(2, 5)

;