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
Return type

Array