torch.Tensor.dim

Tensor.dim() → int

Returns the number of dimensions of self tensor.