quantax.utils.array_set

quantax.utils.array_set(array: Array, inds: Array | ndarray | bool | number | bool | int | float | complex | TypedNdArray, array_set: Array) → Array

Equivalent to array.at[inds].set(array_set), but significantly faster for complex-valued inputs.

Parameters:

• array – The original array.

• inds – The indices to be set.

• array_set – The values to be set.