From 1ae27184acede8b251d5f95acdefa0212eea9081 Mon Sep 17 00:00:00 2001
From: Lou Bernabeu <lou.bernabeu@student-cs.fr>
Date: Tue, 5 Jan 2021 21:12:57 +0100
Subject: [PATCH] forgot a 2
---
elements/main.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/elements/main.py b/elements/main.py
index ddc33e2..60722db 100644
--- a/elements/main.py
+++ b/elements/main.py
@@ -50,9 +50,9 @@ class BeamSplitter(Element):
self.output_vector = self.matrix @ input_state
if side == "left":
- ans, _ = np.split(self.output_vector)
+ ans, _ = np.split(self.output_vector, 2)
elif side == "right":
- _, ans = np.split(self.output_vector)
+ _, ans = np.split(self.output_vector, 2)
return ans
--
GitLab