fixed dag , batch_dag

EQuS · Feb 27, 2024 · 4e69b7e · 4e69b7e
1 parent 3792305
commit 4e69b7e
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/jaxquantum/quantum/base.py b/jaxquantum/quantum/base.py
@@ -78,7 +78,17 @@ def dag(op: jnp.ndarray) -> jnp.ndarray:
     Returns:
         conjugate transpose of op
     """
-    # return jnp.conj(op).T
+    return jnp.conj(op).T
+
+def batch_dag(op: jnp.ndarray) -> jnp.ndarray:
+    """Conjugate transpose.
+
+    Args:
+        op: operator
+
+    Returns:
+        conjugate of op, and transposes last two axes
+    """
     return jnp.moveaxis(jnp.conj(op), -1, -2) # transposes last two axes, good for batching