2021年8月20日
pytorch框架学习之gather()方法的详解
摘要： 给一个张量 索引 index0 index1 index2 index3 index0 0 1 2 3 index1 4 5 6 7 index2 8 9 10 11 index3 12 13 14 15 torch.gather(dim, index) → Tensor 首先dim表示维度，如果d 阅读全文
posted @ 2021-08-20 17:34 coderJ_ONE 阅读(198) 评论(0) 推荐(0) 编辑