torch.stcak()函数对多个张量在维度上进行叠加。 其中参数dim代表不同的维度。 具体如下代码所示: >>> a = torch.ones(3,3) >>> a tensor([[1., 1., 1.], [1., 1., 1.], [1., 1., 1.]]) >>> b = torch.ones(3,3) + 1 >>> b tensor([[2., 2., 2.], [2., 2., 2.], [2., 2., 2.]]) >>> c = torch.ones(3,3) + 2 >>> c tensor([[3.,