Fix normalize

luisfpereira · Dec 20, 2024 · 30e8258 · 30e8258
1 parent 9a50c95
commit 30e8258
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/geomfum/linalg.py b/geomfum/linalg.py
@@ -9,7 +9,7 @@ def _prefix_with_ellipsis(string):
 
 def normalize(array, axis=-1):
     # TODO: handle norm zero?
-    return array / np.linalg.norm(array, axis=1, keepdims=True)
+    return array / np.linalg.norm(array, axis=axis, keepdims=True)
 
 
 def columnwise_scaling(vec, mat):