torch.cat(dim=0) = torch.vstack()torch.cat(dim=1) = torch.hstack()除了torch.stack(),dim=1,dim=2有点怪之外,其他的都是正常操作.
確定! 回上一頁