* (кортеж тензоров тензоров, имя тусклое, Tensor out)

Question

Предположим, у меня есть список памяти list_of_tensors = [tensor1, tensor2, tensor3, tensor4]. Каждый элемент является тензорным пиктором формы (1, 1, 84, 84).

Я хочу объединить этот список тензоров, чтобы получить тензор формы (4, 1, 84, 84). torch.cat(TT, dim=0) наверняка позволит мне это сделать. TT должен быть кортежем тензора, поэтому torch.cat(*list_of_tensors, dim=0) или torch.cat((*list_of_tensors), dim=0) не будут работать.

Как использовать list_of_tensors и torch.cat(???, dim=0) для создания нового тензора формы (4, 1, 84, 84)

Answer 1

Вы можете использовать стек , а удалить лишнее измерение с помощью squeeze

c = (torch.stack(list_of_tensors,dim=1)).squeeze(0)

сейчас c. Форма: (4, 1, 84, 84 )

Вы можете найти объяснение здесь: https://discuss.pytorch.org/t/how-to-turn-a-list-of-tensor-to-tensor/8868/6