¿Cómo obtener la forma de un tensor como una lista de int en Pytorch?

Para obtener la forma de un tensor como una lista en PyTorch, podemos usar dos enfoques. Uno usando el método size() y otro usando el atributo de forma de un tensor en PyTorch. En este breve artículo, vamos a ver cómo utilizar ambos enfoques.

Usando el método de tamaño():

El método size() devuelve el tamaño del tensor propio. El valor devuelto es una subclase de una tupla.

import torch
torch.empty(3, 4, 5).size()

Producción: 

torch.Size([3, 4, 5])

Convertimos esto en una lista usando el método list(). 

Ejemplo:

v = torch.tensor([[1,0],[0,1]])
x = list(v.size())
print(x)

Producción:

[2, 2] 

También puede usar la sesión interactiva de Python como se muestra a continuación:

Usando el atributo de forma:

tensor.shape es un alias de tensor.size(), aunque la forma es un atributo y size() es un método. Para verificar esto podemos ejecutar lo siguiente en la sesión interactiva.

De manera similar, podemos convertir esto en una lista usando el método list(). 

Ejemplo:

import torch
v = torch.tensor([[1,0],[0,1]])
x = list(v.shape)
print(x)

 Producción:

 [2,2]

Alternativamente usando la sesión interactiva: