diff --git a/ndonnx/_funcs.py b/ndonnx/_funcs.py index 47a4c9f..1ebb21c 100644 --- a/ndonnx/_funcs.py +++ b/ndonnx/_funcs.py @@ -550,7 +550,7 @@ def matrix_transpose(x): return _unary(x.dtype._ops.matrix_transpose, x) -def tensordot(x, y, axes=2): +def tensordot(x, y, /, *, axes=2): if (out := x.dtype._ops.tensordot(x, y, axes)) is not NotImplemented: return out if (out := y.dtype._ops.tensordot(x, y, axes)) is not NotImplemented: @@ -640,7 +640,7 @@ def permute_dims(x, axes): ) -def repeat(x, repeats, axis=None): +def repeat(x, repeats, /, *, axis=None): if (out := x.dtype._ops.repeat(x, repeats, axis)) is not NotImplemented: return out raise UnsupportedOperationError(f"Unsupported operand type for repeat: '{x.dtype}'")