flatten vector

QuantumApplicationLab · Nov 21, 2023 · cb921f0 · cb921f0
1 parent 64e65e8
commit cb921f0
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/vqls_prototype/solver/vqls.py b/vqls_prototype/solver/vqls.py
@@ -208,6 +208,9 @@ def construct_circuit(  # pylint: disable=too-many-branches
             # ensure the vector is double
             vector = vector.astype("float64")
 
+            if vector.ndim == 2:
+                vector = vector.flatten()
+
             # create the circuit
             nqbit = int(np.log2(len(vector)))
             self.vector_circuit = QuantumCircuit(nqbit, name="Ub")