operator_action

cuquantum.densitymat.jax.operator_action(

op: Operator,

t: float,

state_in_bufs: Array | Sequence[Array],

params: Array | None = None,

device: Device | None = None,

batch_size: int = 1,

) → Array | List[Array]

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.