[major] refine the interface: add the QuantumState class, state.gate is supported such as state.h

Author: Hanrui-Wang

test/quantumstate_test.py

@@ -0,0 +1,28 @@
+import torchquantum as tq
+import pdb
+import numpy as np
+import torch
+
+if __name__ == '__main__':
+    pdb.set_trace()
+    state = tq.QuantumState(
+        n_wires=4,
+        bsz=1
+    )
+    state.hadamard(wires=1, inverse=True)
+    state.h(wires=2)
+    state.rx(wires=1, params=torch.Tensor([np.pi]))
+
+    state.h(wires=0)
+    state.rx(wires=1, params=0.1)
+
+    state.u1(wires=1, params=[0.2])
+    state.qubitunitary(wires=0, params=[[0,1], [1,0]])
+
+    state.u3(wires=2, params=[0.1, 0.2, 0.3])
+    state.crot(wires=[0,3], params=[0.3, 0.4, 0.5])
+    state.rot(wires=3, params=[0.4, 0.5, 0.6])
+
+    print(state)
+
+

torchquantum/__init__.py

@@ -14,6 +14,7 @@
 from .utils import *
 from .noise_model import *
 from .node import *
+from .states import *
 
 # here we check whether the Qiskit parameterization bug is fixed, if not, a
 # warning message will be printed