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 single axis.

Parameters
Return type

Array