torch:shape

(torch:shape tensor)

Returns the dims list of the tensor's data (the linalg shape), or nil for a scalar tensor (rank 0).