*（Tensors 張量的元組，名稱 dim，Tensor out）

Question

假設我有 memory 列表list_of_tensors = [tensor1, tensor2, tensor3, tensor4] 。 每個元素都是一個形狀為(1, 1, 84, 84)的 pytorch 張量。

我想連接該張量列表以獲得形狀(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

您可以使用stack ，並使用擠壓去除多余的尺寸

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

現在 c.shape 是 (4, 1, 84, 84)

你可以在這里找到解釋： https://discuss.pytorch.org/t/how-to-turn-a-list-of-tensor-to-tensor/8868/6