MatrixOperator#

class cuquantum.densitymat.jax.MatrixOperator(
data: Array | ShapeDtypeStruct,
callback: WrappedTensorCallback | None = None,
grad_callback: WrappedTensorGradientCallback | None = None,
)[source]#

PyTree class for cuDensityMat’s matrix operator.

Methods

__init__(
data: Array | ShapeDtypeStruct,
callback: WrappedTensorCallback | None = None,
grad_callback: WrappedTensorGradientCallback | None = None,
) → None[source]#

Initialize a MatrixOperator object.

Parameters:

  • data – Data specification of the matrix operator. If callback is None, data should be a jax.Array; otherwise, data should be a jax.ShapeDtypeStruct.

  • callback – Forward callback for the matrix operator.

  • grad_callback – Gradient callback for the matrix operator.

copy() → MatrixOperator[source]#

Copy the matrix operator.