jax.lax.index_in_dim

Warning

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

jax.lax.index_in_dim

jax.lax.index_in_dim(operand, index, axis=0, keepdims=True)[source]

Convenience wrapper around slice to perform int indexing.

Parameters

  • operand (Any) –

  • index (int) –

  • axis (int) –

  • keepdims (bool) –

Return type

Any

previous

jax.lax.imag

next

jax.lax.index_take