torch.stack

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

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

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

См. также

torch.cat() объединяет данную последовательность вдоль существующей размерности.

Параметры

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

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

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