Add tdg to operators.py

mit-han-lab · May 29, 2023 · 79cf688 · 79cf688
1 parent dae5ca2
commit 79cf688
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/torchquantum/operators.py b/torchquantum/operators.py
@@ -67,6 +67,7 @@
     "EchoedCrossResonance",
     "ECR",
     "SDG",
+    "TDG",
 ]
 
 
@@ -124,6 +125,7 @@ class Operator(tq.QuantumModule):
         "Reset",
         "EchoedCrossResonance",
         "SDG",
+        "TDG",
     ]
 
     parameterized_ops = [
@@ -1330,12 +1332,25 @@ class sdg(Operation, metaclass=ABCMeta):
     @classmethod
     def _matrix(cls, params):
         return cls.matrix
+
+class tdg(Operation, metaclass=ABCMeta):
+    """Class for TDG Gate."""
+
+    num_params = 0
+    num_wires = 1
+    matrix = mat_dict["tdg"]
+    func = staticmethod(tqf.tdg)
+
+    @classmethod
+    def _matrix(cls, params):
+        return cls.matrix
 
 
 H = Hadamard
 SH = SHadamard
 EchoedCrossResonance = ECR
 SDG = sdg 
+TDG = tdg
 
 
 op_name_dict = {
@@ -1404,4 +1419,5 @@ def _matrix(cls, params):
     "ecr": ECR,
     "echoedcrossresonance": ECR,
     "sdg": sdg,
+    "tdg": tdg,
 }