MatrixOperator#

class cuquantum.densitymat.jax.MatrixOperator(data: Array)[source]#

PyTree class for cuDensityMat’s matrix operator.

Methods

__init__(data: Array) → None[source]#

Initialize a MatrixOperator object.

Parameters:: data – Data buffer of the matrix operator.

Attributes

in_axes#: Return the in_axes PyTree spec for vmapping over the batch dimension (axis 0 of data).