Add implementation

Quantco · Jan 12, 2025 · 775a74d · 775a74d
1 parent 80eefd0
commit 775a74d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/ndonnx/_core/_interface.py b/ndonnx/_core/_interface.py
@@ -345,6 +345,9 @@ def permute_dims(self, x, axes) -> ndx.Array:
     def reshape(self, x, shape, *, copy=None) -> ndx.Array:
         return NotImplemented
 
+    def repeat(self, x, repeats, axis=None) -> ndx.Array:
+        return NotImplemented
+
     def roll(self, x, shift, axis) -> ndx.Array:
         return NotImplemented
 

diff --git a/ndonnx/_core/_shapeimpl.py b/ndonnx/_core/_shapeimpl.py
@@ -66,7 +66,7 @@ def expand_dims(self, x, axis):
             )
         )
 
-    def repeat(self, x, repeats, axis):
+    def repeat(self, x, repeats, axis=None):
         if axis is None:
             x = ndx.reshape(x, [-1])
             axis = 0