Is this page helpful?

Operator#

class cuquantum.densitymat.jax.Operator(
dims: Sequence[int],
simplify: bool | SimplifierConfig = False,
)[source]#

PyTree class for cuDensityMat’s operator.

Methods

__init__(
dims: Sequence[int],
simplify: bool | SimplifierConfig = False,
) → None[source]#

Initialize an Operator object.

Parameters:

  • dims – Hilbert space dimensions.

  • simplify – Controls operator-product simplification at compile time. True uses the default passes; False disables all passes; a SimplifierConfig object selects custom conditions. Opt-in for now: the passes are newly added, their failure mode is a silently different result rather than an error, and their benefit has not been quantified.

append(
op_term: OperatorTerm,
dual: bool = False,
coeff: float | complex | Array = 1.0,
) → None[source]#

Append an operator term to an operator.

Parameters:

  • op_term – Operator term to be appended.

  • dual – Duality of the operator term.

  • coeff – Non-batched coefficient or batched coefficients of the operator term.

Attributes

in_axes#

Return the in_axes PyTree spec for vmapping over the batch dimension.