From 84a6a035ba88f07129b9651df79662467de0f071 Mon Sep 17 00:00:00 2001 From: vinitha-balachandran Date: Thu, 8 Feb 2024 12:06:38 +0800 Subject: [PATCH] added missing docstring --- src/qibotn/eval_qu.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/src/qibotn/eval_qu.py b/src/qibotn/eval_qu.py index 9dc639c..78d84ef 100644 --- a/src/qibotn/eval_qu.py +++ b/src/qibotn/eval_qu.py @@ -5,6 +5,8 @@ from qibo.models import Circuit as QiboCircuit def from_qibo(circuit: QiboCircuit, is_mps: False, psi0=None, method='svd', cutoff=1e-6, cutoff_mode='abs'): + """Create a tensornetwork representation of the circuit""" + nqubits = circuit.nqubits gate_opt = {} if is_mps: @@ -28,6 +30,9 @@ def from_qibo(circuit: QiboCircuit, is_mps: False, psi0=None, method='svd', def init_state_tn(nqubits, init_state_sv): + + """Create a matrixproductstate directly from a dense vector""" + dims = tuple(2 * np.ones(nqubits, dtype=int)) return qtn.tensor_1d.MatrixProductState.from_dense(init_state_sv, dims)