Vid "PyTorch Fundamentals" - What is "ndim" for tensor?

[sunnysaw07]

Hello Everyone,
I'm just checking

Code:

import torch
torch.randn(3,1)

[37Lime]

Hello,

tensor.ndim returns the number of dimensions of the tensor (you can also use tensor.dim())

in your case:

torch.randn(3,1).ndim

would give you 2, the number of dimensions.

Best wishes.