jax.lax.index_take# jax.lax.index_take(src, idxs, axes)[source]# Parameters: src (Array) – idxs (Array) – axes (Sequence[int]) – Return type: Array