gather的作用:根据维度(dim)和索引(index)获取输入(input)的某些特殊值。
根据官方文档中的例子(Example),t是2×2张量:索引是2×2张张量:
按dim=1,也就是列维度取索引值(2×2索引就是0或1代表第一列和第二列),所以根据索引与t的对应关系得到gather后的结果:
解释:根据对应的位置和索引值确定
t的第1列元素并对应索引位置,对应的第1行,是4,第0列元素是3,按照索引的对应位置得到gather后的结果。
gather的作用:根据维度(dim)和索引(index)获取输入(input)的某些特殊值。
根据官方文档中的例子(Example),t是2×2张量:索引是2×2张张量:
按dim=1,也就是列维度取索引值(2×2索引就是0或1代表第一列和第二列),所以根据索引与t的对应关系得到gather后的结果:
解释:根据对应的位置和索引值确定
t的第1列元素并对应索引位置,对应的第1行,是4,第0列元素是3,按照索引的对应位置得到gather后的结果。