From 08a1b2a2d747d67c047c754d38b87268dd0fded5 Mon Sep 17 00:00:00 2001 From: Tushar Garg <139030529+tring27@users.noreply.github.com> Date: Tue, 12 Mar 2024 13:54:00 -0700 Subject: [PATCH] Update gate_wrapper.py Bug in U2 gate parameter initialization #232 --- torchquantum/functional/gate_wrapper.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torchquantum/functional/gate_wrapper.py b/torchquantum/functional/gate_wrapper.py index f1383f2f..c446df6b 100644 --- a/torchquantum/functional/gate_wrapper.py +++ b/torchquantum/functional/gate_wrapper.py @@ -371,7 +371,7 @@ def gate_wrapper( params = params.unsqueeze(0) if params.dim() == 2 else params else: if params.dim() == 1: - params = params.unsqueeze(-1) + params = params.unsqueeze(0).unsqueeze(-1) elif params.dim() == 0: params = params.unsqueeze(-1).unsqueeze(-1) # params = params.unsqueeze(-1) if params.dim() == 1 else params