python - 使用pytorch获取形状(4,1,84,84)

Question

假设我有四个 pytorch 张量(tensor1、tensor2、tensor3、tensor4)。每个张量的形状为(1, 1, 84, 84)。第一个维度是张量的数量，第二个维度是颜色的数量(例如我们示例中的灰度)，最后两个维度表示图像的高度和宽度。

我想将它们堆叠起来以获得形状(4, 1, 84, 84)。

我尝试了torch.stack((tensor1,tensor2,tensor3,tensor4),dim=0)，但我得到了一个形状(4,1,1,84,84) 。

如何堆叠这些张量以使形状为 (4, 1, 84, 84)

Answer 1

您可以使用连接函数:

a = torch.ones(1,1,84,84)
b = torch.ones(1,1,84,84)
c = torch.cat((a,b), 0) # size[2,1,84,84]