根据得到的索引在输入中取值#[1,2],[3,2]
原理解释
假设输入与上同;index=B;输出为C B中每个元素分别为b(0,0)=0,b(0,1)=0 b(1,0)=1,b(1,1)=0
如果dim...=0(列) 则取B中元素的列号,如:b(0,1)的1 b(0,1)=0,所以C中的c(0,1)=输入的(0,1)处元素2
如果dim=1(行) 则取B中元素的列号,如:b(0,1)的0 b(0,1)=0...,所以C中的c(0,1)=输入的(0,0)处元素1
总结如下:输出 元素 在 输入张量 中的位置为:输出元素位置取决于同位置的index元素 dim=1时,取同位置的index元素的行号做行号,...该位置处index元素做列号 dim=0时,取同位置的index元素的列号做列号,该位置处index元素做行号。...最后根据得到的索引在输入中取值
index类型必须为LongTensor gather最终的输出变量与index同形。