Склейка вдоль существующей оси в PyTorch
Функция cat объединяет список
тензоров в один. Все части должны
совпадать по размерам на остальных
осях; склейка идёт вдоль уже
существующей оси, которую указывают
параметром dim.
Две таблицы с одинаковым числом столбцов можно собрать в одну, дописав ряды друг под другом. Создадим фрагменты и склеим их:
import torch
top = torch.tensor([[1, 2], [3, 4]])
bottom = torch.tensor([[5, 6]])
merged = torch.cat([top, bottom], dim=0)
print(merged)
Тот же приём работает вдоль столбцов: ряды совпадают по длине, а ширина результата суммируется:
import torch
left = torch.tensor([[1, 2], [3, 4]])
right = torch.tensor([[5, 6], [7, 8]])
wide = torch.cat([left, right], dim=1)
print(wide) # выведет tensor([[1, 2, 5, 6], [3, 4, 7, 8]])
Создайте таблицы [[1], [2]]
и [[3], [4]] и выведите
одну таблицу из четырёх рядов
по одному числу.
Создайте ряды [10, 11]
и [12, 13], оформите каждый
как таблицу с одним рядом
и объедините их в таблицу
2 на 2.
Создайте таблицу [[1, 2], [3, 4]]
и ряд [5, 6] как таблицу
с одним рядом; соберите слева
и справа и выведите число
элементов в результате.
