cuquantum.densitymat.jax.MatrixOperator

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

PyTree class for cuDensityMat’s matrix operator.

Methods

__init__(data: jax.Array, callback: cuquantum.bindings.cudensitymat.WrappedTensorCallback | None = None, grad_callback: cuquantum.bindings.cudensitymat.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.