torch.stack

torch.stack(tensors, dim=0, *, out=None) → Tensor

Объединяет последовательность тензоров вдоль новой размерности.

Все тензоры должны иметь одинаковый размер.

Параметры:

• tensors (последовательность тензоров) – последовательность тензоров для объединения
• dim (int) – размерность для вставки. Должна быть между 0 и количеством измерений объединенных тензоров (включительно)

Ключевые аргументы:

out (Тензор, необязательно) – выходной тензор.