Merge pull request #27 from yangguohao/task76

【PaddlePaddle Hackathon】76 量子电路的量子比特数扩展

paddle_quantum/circuit.py

@@ -26,7 +26,7 @@ from paddle import imag, real, reshape, kron, matmul, trace
 from paddle_quantum.utils import partial_trace, dagger, pauli_str_to_matrix
 from paddle_quantum import shadow
 from paddle_quantum.intrinsic import *
-from paddle_quantum.state import density_op
+from paddle_quantum.state import density_op,vec
 
 __all__ = [
     "UAnsatz",
@@ -59,7 +59,30 @@ class UAnsatz:
                         paddle.to_tensor(np.array([math.pi / 4])), paddle.to_tensor(np.array([-math.pi / 4]))]
         # Record history of adding gates to the circuit
         self.__history = []
-
+        
+    def expand(self,new_n):
+        """
+        为原来的量子电路进行比特数扩展
+
+        Args：
+            new_n(int):扩展后的量子比特数
+        """
+        assert new_n>=self.n,'扩展后量子比特数要大于原量子比特数'
+        diff = new_n-self.n
+        dim = 2**diff
+        if self.__state is not None:
+            if self.__run_mode=='density_matrix':
+                shape = (dim,dim)
+                _state = paddle.to_tensor(density_op(diff))
+            elif self.__run_mode=='state_vector':
+                shape = (dim,)
+                _state = paddle.to_tensor(vec(0,diff))
+            
+            _state= paddle.reshape(_state,shape)
+            _state = kron(self.__state,_state)
+            self.__state = _state
+        self.n = new_n
+        
     def __add__(self, cir):
         r"""重载加法 ‘+’ 运算符，用于拼接两个维度相同的电路
 

test_and_documents/readme.md

+通过在UAnsatz类中添加新的成员函数expand来实现扩展

test_and_documents/test.py

+from paddle_quantum.circuit import UAnsatz
+from paddle import kron
+from paddle_quantum.state import vec,density_op
+import paddle
+
+
+#density_matrix
+def test_density_matrix():
+    cir = UAnsatz(1)
+    cir.ry(paddle.to_tensor(1,dtype='float64'),0)
+    state = cir.run_density_matrix()
+    cir.expand(3)
+    print(cir.get_state())
+
+    cir2 = UAnsatz(3)
+    cir2.ry(paddle.to_tensor(1,dtype='float64'),0)
+    cir2.run_density_matrix()
+    print(cir2.get_state())
+
+#state_vector
+def test_state_vector():
+    cir = UAnsatz(1)
+    cir.ry(paddle.to_tensor(1,dtype='float64'),0)
+    state = cir.run_state_vector()
+    cir.expand(3)
+    print(cir.get_state())
+
+    cir2 = UAnsatz(3)
+    cir2.ry(paddle.to_tensor(1,dtype='float64'),0)
+    cir2.run_state_vector()
+    print(cir2.get_state())
+
+test_density_matrix()
+test_state_vector()