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 星辰之衍 阅读(7031) 评论(0) 推荐(1) 编辑

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

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

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

导航