torch.gather torch.gather(input,dim,index) 从input矩阵中按照特定的下标选值,最终矩阵大小和指定的下标矩阵大小相同 dim指定方向:dim=1意味自己指定行,而列的下标从index矩阵中得到,vice versa. import torch t = torch.tensor([[1, 2], [3, 4]]