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]