根据得到的索引在输入中取值#[1,1],[4,3]
c = torch.gather(a,0,torch.LongTensor([[0,0],[1,0]]))#1....=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同形。
云服务器
ICP备案
对象存储
腾讯会议
云直播
