Función torch.eye de PyTorch
Manual de referencia de Pytorch torch
torch.eyees una función en PyTorch que se utiliza para crear una matriz identidad (una matriz con elementos en la diagonal iguales a 1 y el resto iguales a 0).
Esto se usa comúnmente en el aprendizaje profundo para escenarios como inicialización, creación de máscaras, etc.
Definición de la función
torch.eye(n, m, dtype, device, requires_grad)
Parámetros:
n(int): número de filas.m(int, opcional): número de columnas. Si no se especifica, se crea una matriz cuadrada de n×n.dtype(torch.dtype, opcional): tipo de datos.device(torch.device, opcional): dispositivo.requires_grad(bool, opcional): si se necesita calcular gradientes.
Valor de retorno:
torch.Tensor: Devuelve la matriz identidad.
Ejemplos de uso
Ejemplo 1: Crear una matriz cuadrada
Ejemplo
import torch
# Crear una matriz identidad de 3x3
I = torch.eye(3)
print(I)
# Crear una matriz identidad de 3x3
I = torch.eye(3)
print(I)
El resultado es:
tensor([[1., 0., 0.],
[0., 1., 0.],
[0., 0., 1.]])
Ejemplo 2: Crear una matriz no cuadrada
Ejemplo
import torch
# Crear una matriz identidad de 3x4
I = torch.eye(3, 4)
print(I)
# Crear una matriz identidad de 3x4
I = torch.eye(3, 4)
print(I)
El resultado es:
tensor([[1., 0., 0., 0.],
[0., 1., 0., 0.],
[0., 0., 1., 0.]])
Otras extensiones