added dag for data in qarray

EQuS · Mar 4, 2024 · aa4b2d9 · aa4b2d9
1 parent a0c06ed
commit aa4b2d9
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/jaxquantum/core/qarray.py b/jaxquantum/core/qarray.py
@@ -453,4 +453,15 @@ def batch_dag_data(op: Array) -> Array:
     """
     return jnp.moveaxis(
         jnp.conj(op), -1, -2
-    )  # transposes last two axes, good for batching
+    )  # transposes last two axes, good for batching
+
+def dag_data(op: Array) -> Array:
+    """Conjugate transpose.
+
+    Args:
+        op: operator
+
+    Returns:
+        conjugate of op, and transposes last two axes
+    """
+    return jnp.conj(op.T)