fixed repeate_interleave key error

dhanush-2501 · Sep 12, 2023 · 8f22d4e · 8f22d4e
1 parent 637bbd9
commit 8f22d4e
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/ivy/functional/frontends/paddle/manipulation.py b/ivy/functional/frontends/paddle/manipulation.py
@@ -77,6 +77,15 @@ def gather(params, indices, axis=-1, batch_dims=0, name=None):
     return ivy.gather(params, indices, axis=axis, batch_dims=batch_dims)
 
 
+@with_supported_dtypes(
+    {"2.5.1 and below": ("int32", "int64", "float32", "float64")},
+    "paddle",
+)
+@to_ivy_arrays_and_back
+def repeat_interleave(x, repeats, axis=None, name=None):
+    return ivy.repeat(x, repeats, axis=axis)
+
+
 @to_ivy_arrays_and_back
 def reshape(x, shape):
     return ivy.reshape(x, shape)