Warning

This page was created from a pull request.

jax.lax.dynamic_update_slice¶

jax.lax.dynamic_update_slice(operand, update, start_indices)[source]¶

Wraps XLA’s DynamicUpdateSlice operator.

Parameters

  • operand (Any) – an array to slice.

  • update (Any) – an array containing the new values to write onto operand.

  • start_indices (Any) – a list of scalar indices, one per dimension.

Return type

Any

Returns

An array containing the slice.