2023年1月13日
pytorch gather函数
摘要： pytorch.gather eg 四个输出 分别为 1 1 3 3 if dim=0 3,4,1,2 阅读全文
posted @ 2023-01-13 14:45 夜未央111 阅读(9) 评论(0) 推荐(0) 编辑