pytorch 函数

 

 

x=torch.tensor([[1,2,3],

　　　　　　　[4,5,6]])

 

torch.gather(dim=0,index=torch.tensor([0,0,1],[1,0,1])) 表示提取索引元素，索引由index提供的信息生成）（索引！=index）

dim=0,index里的元素表示 索引的第0维(行号) ，然后根据index的第一维（列序号）添加为索引的第一维

索引=0,0   0,1  1,2   1,0   0,1  1,2  

输出结果与index矩阵一致

结果：[1,2,6] ,[4,2,6]