Функция PyTorch torch.eye

torch.eye— это функция в PyTorch, используемая для создания единичной матрицы (матрицы с 1 на диагонали и 0 в остальных позициях).

Обычно она применяется в глубоком обучении для инициализации, создания масок и в других сценариях.

torch.eye(n, m, dtype, device, requires_grad)

Parameters:

n(int): количество строк.
m(int, необязательно): количество столбцов. Если не указано, создаётся квадратная матрица n×n.
dtype(torch.dtype, необязательно): тип данных.
device(torch.device, необязательно): устройство.
requires_grad(bool, необязательно): требуется ли вычисление градиента.

Возвращаемое значение:

Примеры использования

import torch

# Создать единичную матрицу 3x3
I = torch.eye(3)

print(I)

Вывод:

tensor([[1., 0., 0.],
        [0., 1., 0.],
        [0., 0., 1.]])

import torch

# Создать единичную матрицу 3x4
I = torch.eye(3, 4)

print(I)

Вывод:

tensor([[1., 0., 0., 0.],
        [0., 1., 0., 0.],
        [0., 0., 1., 0.]])

Другие расширения