2022 年 11月 5 日随笔档案 - 村雨sup

摘要： a.shape() = (2,3) b.shape() = (2,3) torch.cat([a,b], dim=0).shape() = (4,3) torch.cat([a,b], dim=1).shape() = (2,6) 阅读全文

posted @ 2022-11-05 21:51 村雨sup 阅读(41) 评论(0) 推荐(0) 编辑