torch.cat,torch.stack,torch.vstack,torch.hstack
torch.cat(dim=0) = torch.vstack()
猜你喜欢
转载自blog.csdn.net/u010590593/article/details/116596869
今日推荐
周排行