Единичная матрица в PyTorch
Единичную матрицу удобно получить
через eye: квадратная таблица,
на главной диагонали стоят единицы,
а в остальных ячейках нули.
Создадим единичную матрицу размером
3 на 3 и выведем её:
import torch
unit = torch.eye(3)
print(unit) # выведет tensor([[1., 0., 0.], [0., 1., 0.], [0., 0., 1.]])
Создайте единичную матрицу размером
2 на 2 и выведите её.
Создайте единичную матрицу размером
4 на 4 и выведите её.
Создайте единичную матрицу размером
3 на 3 и выведите её.
