Similar operation like `numpy.take`

[Deepali (Deepali Patel)]

Do you need something like - Select rows from a 2D tensor