Avoid flatten() so as to keep the shape information of contraction results

2023-04-19 16:23:59 +08:00
parent 4d36afb9ef
commit e517b4fe7c
2 changed files with 3 additions and 2 deletions
--- a/src/qibotn/cutn.py
+++ b/src/qibotn/cutn.py
@@ -7,4 +7,4 @@ def eval(qibo_circ, datatype):
    myconvertor = QiboCircuitToEinsum(qibo_circ, dtype=datatype)
    operands_expression = myconvertor.state_vector()
    results = contract(*operands_expression)
-    return results.flatten()
+    return results