diff --git a/elements/main.py b/elements/main.py index ddc33e25797f36ec6f4574055e65a34969be05c7..60722db37732ad6d5b35865efea4f5906e1fc57b 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