Function take

Take elements from an array along an axis.

This is equivalent to advanced indexing with integer indices over that numbered axis. By default, the flattened array is used.