torch.index_select — PyTorch 1.10.1 documentation
pytorch.org › docs › stabletorch.index_select — PyTorch 1.10.0 documentation torch.index_select torch.index_select(input, dim, index, *, out=None) → Tensor Returns a new tensor which indexes the input tensor along dimension dim using the entries in index which is a LongTensor. The returned tensor has the same number of dimensions as the original tensor ( input ).