2018年8月9日

Pytorch的gather用法理解

摘要： 先放一张表，可以看成是二维数组 |行(列)索引|索引0| 索引1| 索引2|索引3| | : :| : : | : : | : : | : : | |索引0 |0|1|2|3| |索引1 |4|5|6|7| |索引2 |8|9|10|11| |索引3 |12|13|14|15| 看一下下面例子代码： 阅读全文

posted @ 2018-08-09 21:32 星辰之衍 阅读(21300) 评论(1) 推荐(3) 编辑

关于Pytorch的二维tensor的gather和scatter_操作用法分析

摘要： 看得不明不白（我在下一篇中写了如何理解gather的用法） gather是一个比较复杂的操作，对一个2维tensor，输出的每个元素如下： 二维tensor的gather操作 针对0轴 注意index此时的值 输入 输出 ======================分割线============== 阅读全文

posted @ 2018-08-09 19:36 星辰之衍 阅读(5939) 评论(0) 推荐(0) 编辑

导航