用药剂获得形状（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]