fix(ivy): Fix ivy.dft to avoid slice object containing scalar array

ivy-llc · Sep 25, 2023 · 9ee0f94 · 9ee0f94
1 parent 78e8f99
commit 9ee0f94
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ivy/functional/ivy/experimental/layers.py b/ivy/functional/ivy/experimental/layers.py
@@ -1320,7 +1320,7 @@ def dft(
 
     if onesided:
         slices = [slice(0, a) for a in res.shape]
-        slices[axis] = slice(0, ivy.shape(res, as_array=True)[axis] // 2 + 1)
+        slices[axis] = slice(0, ivy.shape(res)[axis] // 2 + 1)
         res = res[tuple(slices)]
     return res