index_select

dragon.vm.torch.index_select(
  input,
  dim,
  index,
  out=None
)[source]

Select the elements along the given dim using index.

Parameters:
Returns:

dragon.vm.torch.Tensor – The output tensor.