From 61edc12522c1075547044fba85df35faa3a14bdd Mon Sep 17 00:00:00 2001
From: franksim <simon.franken@uni-a.de>
Date: Mon, 20 Oct 2025 13:04:19 +0000
Subject: [PATCH] main: fix

---
 mmp/a1/tensors.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mmp/a1/tensors.py b/mmp/a1/tensors.py
index 72e1c29..4c926c6 100644
--- a/mmp/a1/tensors.py
+++ b/mmp/a1/tensors.py
@@ -14,4 +14,4 @@ def mask(foreground: torch.Tensor, background: torch.Tensor, mask_tensor: torch.
 
 
 def add_matrix_vector(matrix: torch.Tensor, vector: torch.Tensor):
-    return matrix.add(vector)
+    return matrix + vector.unsqueeze(1)