torch.stack, torch.cat用法详解 针对二维数据dim=0,1,2dim=0dim=1dim=2 针对三维数据dim=0,1,2,3dim=0或1 是一样的dim=2dim=3 torch.cat() 针对二维数据dim=0,1,2 dim=0 dim=1 dim=2 针对三维数据dim=0,1,2,3 dim=0或1 是一样的 dim=2 dim=3 所以可以看出dim是根据数据的维度从最后一维往前取d,d-1,d-2 其中 d-n(n>2) -> d=0都与d-2相同。 torch.cat()