From 69ba8d54b347be5deaae40f065f2d5f24df7d034 Mon Sep 17 00:00:00 2001 From: Austin Huang <65315367+austingmhuang@users.noreply.github.com> Date: Wed, 24 Jul 2024 14:20:24 -0400 Subject: [PATCH] Update pennylane/registers.py Co-authored-by: Utkarsh --- pennylane/registers.py | 16 +++++++++++----- 1 file changed, 11 insertions(+), 5 deletions(-) diff --git a/pennylane/registers.py b/pennylane/registers.py index 286fe50a774..696dcd5a70c 100644 --- a/pennylane/registers.py +++ b/pennylane/registers.py @@ -60,17 +60,23 @@ def registers(register_dict): .. code-block:: - reg = qml.registers({"aux": 1, "phi": 5, "psi": 5}) + dev = qml.device("default.qubit") + reg = registers({"aux": 1, "phi": 5, "psi": 5}) + @qml.qnode(dev) def circuit(): - qml.Hadamard(reg["aux"]) + for state in ["phi", "psi"]: + qml.BasisState([1, 1, 0, 0, 0], reg[state]) + qml.Hadamard(reg["aux"]) for i in range(len(reg["phi"])): - qml.CSWAP(reg["aux"] | reg["phi"][i] | reg["psi"][i]) - + qml.CSWAP(reg["aux"] + reg["phi"][i] + reg["psi"][i]) qml.Hadamard(reg["aux"]) - + return qml.expval(qml.Z(wires=reg["aux"])) + +>>> circuit() +0.9999999999999996 """ def _registers(register_dict, _start_wire_index=0):