Функция eye
Функция eye создаёт двумерный тензор,
представляющий единичную матрицу. На главной диагонали
такого тензора располагаются единицы, а все остальные
элементы равны нулю. Первым обязательным параметром
функция принимает количество строк (n),
вторым - количество столбцов (m).
Если указан только один параметр, создаётся квадратная
матрица. Также можно задать тип данных элементов
с помощью параметра dtype и устройство
размещения через device.
Синтаксис
torch.eye(n, [m], [dtype], [device])
Пример
Давайте создадим единичную матрицу размером
3 на 3:
import torch
t = torch.eye(3)
print(t)
Результат выполнения кода:
tensor([
[1., 0., 0.],
[0., 1., 0.],
[0., 0., 1.]
])
Пример
Теперь создадим прямоугольную матрицу с
3 строками и 4 столбцами:
import torch
t = torch.eye(3, 4)
print(t)
Результат выполнения кода:
tensor([
[1., 0., 0., 0.],
[0., 1., 0., 0.],
[0., 0., 1., 0.]
])
Пример
Создадим единичную матрицу с элементами
типа int:
import torch
t = torch.eye(3, dtype=torch.int)
print(t)
Результат выполнения кода:
tensor([
[1, 0, 0],
[0, 1, 0],
[0, 0, 1]
])
