Один ряд элементов в PyTorch

Все элементы таблицы или другого многомерного тензора можно записать подряд в один ряд. Для этого служит метод flatten. Порядок чисел сохраняется так, как они шли в памяти до свёртки.

Создадим таблицу из четырёх чисел и выведем результат flatten:

import torch

block = torch.tensor([[1, 2], [3, 4]])
line = block.flatten()
print(line) # выведет tensor([1, 2, 3, 4])

Форма после вызова всегда одномерная. Длина ряда равна общему числу элементов исходного тензора:

import torch

block = torch.tensor([[1, 2], [3, 4]])
line = block.flatten()
print(line.shape) # выведет torch.Size([4])

Соберите таблицу из рядов [5, 6] и [7, 8] и выведите все её ячейки одним рядом из четырёх чисел.

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

Соберите ряд из чисел 1, 2 и 3, сверните его в один ряд и выведите полученные значения.