Fix 1x1 dimension matrix simplification

LukasZahradnik · Oct 29, 2023 · b0ad5c4 · b0ad5c4
1 parent fc3ffb6
commit b0ad5c4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/neuralogic/core/constructs/java_objects.py b/neuralogic/core/constructs/java_objects.py
@@ -36,7 +36,7 @@ def get_value(self, weight):
                     value = self.vector_value(weight[0])
             elif len(weight) == 2:
                 if weight[0] == 1:
-                    value = self.vector_value(weight[1], True)
+                    value = self.scalar_value() if weight[1] == 1 else self.vector_value(weight[1], True)
                 elif weight[1] == 1:
                     value = self.vector_value(weight[0], False)
                 else: