cuquantum.densitymat.jax.Operator

class cuquantum.densitymat.jax.Operator(dims: Sequence[int])[source]

PyTree class for cuDensityMat’s operator.

Methods

__init__(dims: Sequence[int]) → None[source]

Initialize an Operator object.

Parameters

dims – Hilbert space dimensions.

append(op_term: pysrc.operators.OperatorTerm, *, dual: bool = False, coeff: float = 1.0, coeff_callback: cuquantum.bindings.cudensitymat.WrappedScalarCallback | None = None, coeff_grad_callback: cuquantum.bindings.cudensitymat.WrappedScalarGradientCallback | None = None) → None[source]

Append an operator term to an operator.

Parameters

  • op_term – Operator term to be appended.

  • dual – Duality of the operator term.

  • coeff – Coefficient of the operator term.

  • coeff_callback – Forward callback for the coefficient.

  • coeff_grad_callback – Gradient callback for the coefficient.