jax.lax.index_take

Warning

This page was created from a pull request (#9655).

jax.lax.index_take

jax.lax.index_take(src, idxs, axes)[source]
Parameters
Return type

Any

previous

jax.lax.index_in_dim

next

jax.lax.iota