Транспонирование таблицы в PyTorch

У двумерной таблицы метод transpose меняет местами две выбранные оси. Для обычной матрицы из рядов и столбцов это транспонирование: бывшие ряды становятся столбцами.

Создадим таблицу из двух рядов по три числа и поменяем местами оси с индексами 0 и 1:

import torch

matrix = torch.tensor([[1, 2, 3], [4, 5, 6]])
flipped = matrix.transpose(0, 1)
print(flipped) # выведет tensor([[1, 4], [2, 5], [3, 6]])

Форма после обмена отражает новые длины осей: было два ряда по три ячейки, стало три ряда по две:

import torch

matrix = torch.tensor([[1, 2, 3], [4, 5, 6]])
flipped = matrix.transpose(0, 1)
print(flipped.shape) # выведет torch.Size([3, 2])

Соберите таблицу из рядов [1, 2] и [3, 4] и выведите её после обмена рядов и столбцов местами.

Соберите таблицу из трёх рядов по два числа и выведите описание формы после обмена первой и второй оси.

Соберите таблицу из одного ряда [10, 20, 30, 40] и выведите результат обмена осей 0 и 1.