Cambiar vista de Tensor en Pytorch

En este artículo, vamos a cambiar la vista del tensor dado en PyTorch. Para esto, usaremos la función view() para cambiar el tensor en filas y columnas de IE de formato bidimensional. Tenemos que especificar el número de filas y el número de columnas a visualizar.

Sintaxis: tensor.view(no_of_rows,no_of_columns)

Ejemplo 1: programa en Python para crear un tensor con 10 elementos y vista con 5 filas y 2 columnas y viceversa

# importing torch module
import torch
  
# create one dimensional tensor 
# 10 elements
a = torch.FloatTensor([10, 20, 30, 40, 50, 1, 2, 3, 4, 5])  
  
# view tensor in 5 rows and 2
# columns
print(a.view(5, 2))
   
# view tensor in 2 rows and 5 
# columns
print(a.view(2, 5))

Producción:

Ejemplo 2: Cambiar la vista de un tensor a 4 filas y 3 columnas y viceversa

# importing torch module
import torch
  
# create one dimensional tensor 12 elements
a = torch.FloatTensor([34, 56, 10, 20, 30, 40, 50, 1, 2, 3, 4, 5])
  
# view tensor in 4 rows and 3 columns
print(a.view(4, 3))
  
# view tensor in 3 rows and 4 columns
print(a.view(3, 4))

Producción: