Как я могу чередовать 5 тензоров PyTorch?

Question

У меня есть 5 тензоров формы torch.Size([7, 20, 180])

Я хочу чередовать их один за другим вдоль dim=1. Так что моя окончательная форма будет torch.Size([7, 100, 180]).

По сути, я хочу первый элемент из первого тензора, затем первый элемент из второго тензора и т. Д.

Answer 1

Если я правильно понял,

import torch
stacked = torch.stack(list_of_tensors, dim=2)
interleaved = torch.flatten(stacked, start_dim=1, end_dim=2)

interleaved - это то, что вам нужно, по-видимому

(протестировано с pytorch 1.1.0)