cuquantum.densitymat.jax.operator_action

cuquantum.densitymat.jax.operator_action(op: pysrc.operators.Operator, t: float, state_in_bufs: Union[jax.Array, Sequence[jax.Array]], params: jax.Array | None = None, device: jaxlib._jax.Device | None = None, batch_size: int = 1) → Union[jax.Array, List[jax.Array]][source]

Compute the action of an operator on a state.

Parameters

  • op – Operator to compute the action of.

  • t – The time to compute the operator action at.

  • state_in_bufs – Buffers of the input state components.

  • params – Callback parameters used to construct the operator.

  • device – Device to use for the operator action.

  • batch_size – Batch size of the operator action.

Returns

Buffers of the output state components.