jax.lax.dynamic_update_slice_in_dim

jax.lax.dynamic_update_slice_in_dim(operand, update, start_index, axis)

Convenience wrapper around dynamic_update_slice() to update a slice in a single axis.

Parameters

• operand (Any) –

• update (Any) –

• start_index (Any) –

• axis (int) –

Return type

Any