ElementaryOperator#

class cuquantum.densitymat.jax.ElementaryOperator(
data: Array,
diag_offsets: Sequence[int] = (),
)[source]#

PyTree class for cuDensityMat’s elementary operator.

Methods

__init__(
data: Array,
diag_offsets: Sequence[int] = (),
) → None[source]#

Initialize an ElementaryOperator object.

Parameters:

  • data – Data buffer of the elementary operator.

  • diag_offsets – Diagonal offsets of the elementary operator.

Attributes

in_axes#

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