torch.cat

 

 

torch.cat

 

 用于连接多个张量

当dim=0时张量按竖直方向连接，dim=1时按水平方向连接

 

posted @ 2021-07-28 11:50  祥瑞哈哈哈  阅读(121)  评论(0)  编辑  收藏  举报