MatrixOperator#

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

PyTree class for cuDensityMat’s matrix operator.

Methods

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

Initialize a MatrixOperator object.

Parameters:

data – Data buffer of the matrix operator.
callback – Forward callback for the matrix operator.
grad_callback – Gradient callback for the matrix operator.