Умножение матриц в PyTorch
Две таблицы из двух измерений перемножают
оператором @ по правилам
умножения матриц. Число столбцов
первой таблицы должно совпадать
с числом строк второй; результат
имеет столько строк, сколько у первой
таблицы, и столько столбцов, сколько
у второй.
Создадим две квадратные таблицы
2×2 и перемножим их:
import torch
A = torch.tensor([[1, 2], [3, 4]])
B = torch.tensor([[5, 6], [7, 8]])
C = A @ B
print(C) # выведет tensor([[19, 22], [43, 50]])
Строка из одного ряда и столбец согласуются, если длина ряда равна числу строк столбца:
import torch
row = torch.tensor([[1, 2, 3]])
col = torch.tensor([[4], [5], [6]])
D = row @ col
print(D) # выведет tensor([[32]])
Соберите таблицы [[1, 0], [0, 1]]
и [[2, 3], [4, 5]], получите
их матричное произведение и выведите его.
Соберите ряд [[1, 2, 3]] и столбец
[[1], [0], [1]], перемножьте
их как матрицы и выведите результат.
Соберите таблицы [[2, 0], [1, 1]]
и [[1, 2], [3, 4]], получите
итоговую таблицу размером 2×2
и выведите её.
