torch.cat,torch.stack,torch.vstack,torch.hstack

torch.cat(dim=0) = torch.vstack()
torch.cat(dim=1) = torch.hstack()

除了torch.stack(),dim=1,dim=2有点怪之外，其他的都是正常操作

高版本torch不再支持torch.hstack和torch.vstack