From 06d9bb5322f03b01c366190650a788a8dceade5b Mon Sep 17 00:00:00 2001
From: NAUD Maxence <maxence.naud@cea.fr>
Date: Fri, 19 Apr 2024 15:31:15 +0000
Subject: [PATCH] fix bias dim in test_recipes.py
---
aidge_backend_cpu/unit_tests/test_recipes.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/aidge_backend_cpu/unit_tests/test_recipes.py b/aidge_backend_cpu/unit_tests/test_recipes.py
index 5586ab24..12d87743 100644
--- a/aidge_backend_cpu/unit_tests/test_recipes.py
+++ b/aidge_backend_cpu/unit_tests/test_recipes.py
@@ -40,7 +40,7 @@ class test_recipes(unittest.TestCase):
graph_view.set_backend("cpu")
np_weights = np.arange(9).reshape([1, 1, 3, 3]).astype(np.float32)
- np_bias = np.arange(1).reshape([1, 1]).astype(np.float32)
+ np_bias = np.arange(1).reshape([1]).astype(np.float32)
np_scale = np.array([0.05]).astype(np.float32)
np_shift = np.array([0.05]).astype(np.float32)
--
GitLab