dpnp.choose

dpnp.choose(x1, choices, out=None, mode='raise')

Construct an array from an index array and a set of arrays to choose from.

For full documentation refer to numpy.choose.

See also

dpnp.take_along_axis

Preferable if choices is an array.