Determinar el número de dimensiones de un Tensor en pytorch

• 1. En circunstancias normales, la dimensión de un tensor se  juzga por tensor.shape  , por ejemplo:

import torch

x = torch.randint(low=-10,high=10,size=(3,6))
print(x)
print(x.shape)

 

A partir de esto, se puede juzgar que es bidimensional (3 elementos en la primera dimensión y 6 elementos en la segunda dimensión).

• 2. Pero a veces, es necesario juzgar si un Tensor es unidimensional, bidimensional o tridimensional, etc., en el código ¿Cómo juzgar? Se puede juzgar por len(tensor.shape) , por ejemplo:

import torch

x = torch.randint(low=-10,high=10,size=(3,6))
print(x)
print(x.shape)
print(len(x.shape))

 

La dimensión de salida será 2.