Método Python Pytorch eye()

PyTorch es una biblioteca de aprendizaje automático de código abierto desarrollada por Facebook. Se utiliza para fines de procesamiento de lenguaje natural y redes neuronales profundas.

La función torch.eye()devuelve a devuelve un tensor 2-D de tamaño n*m con unos en la diagonal y ceros en el resto.

Sintaxis : torch.eye(n, m, out=Ninguno)

Parámetros :
n : el número de filas
m : el número de columnas. Predeterminado – n
out (Tensor, opcional) : el tensor de salida

Tipo de retorno : un tensor 2-D

Código #1:

# Importing the PyTorch library
import torch
  
  
# Applying the eye function and
# storing the resulting tensor in 'a'
a = torch.eye(3, 4)
print("a = ", a)
  
b = torch.eye(3, 3)
print("b = ", b)
  
c = torch.eye(5, 1)
print("c = ", c)

Producción:

a =  tensor([[1., 0., 0., 0.],
        [0., 1., 0., 0.],
        [0., 0., 1., 0.]])
b =  tensor([[1., 0., 0.],
        [0., 1., 0.],
        [0., 0., 1.]])
c =  tensor([[1.],
        [0.],
        [0.],
        [0.],
        [0.]])