jax.lax.dynamic_update_slice_in_dim
jax.lax.dynamic_update_slice_in_dim#
- jax.lax.dynamic_update_slice_in_dim(operand, update, start_index, axis)[source]#
Convenience wrapper around
dynamic_update_slice()
to update a slice in a singleaxis
.
Convenience wrapper around dynamic_update_slice()
to update a slice
in a single axis
.