jax.lax.dynamic_slice_in_dim

Warning

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

jax.lax.dynamic_slice_in_dim

jax.lax.dynamic_slice_in_dim(operand, start_index, slice_size, axis=0)[source]

Convenience wrapper around dynamic_slice applying to one dimension.

Parameters

  • operand (Any) –

  • start_index (Any) –

  • slice_size (int) –

  • axis (int) –

Return type

Any

previous

jax.lax.dynamic_slice

next

jax.lax.dynamic_update_slice