用药剂获得形状（4，1，84，84）

Question

假设我有四个拜托奇张量。每个张量都是形状的 。第一个维度是张量的数量，第二个维度是颜色的数量（例如，在我们的示例中是灰度），最后两个维度表示图像的高度和宽度。(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]