diff --git a/ivy/data_classes/container/experimental/layers.py b/ivy/data_classes/container/experimental/layers.py index 982411992e442..c1629291138ec 100644 --- a/ivy/data_classes/container/experimental/layers.py +++ b/ivy/data_classes/container/experimental/layers.py @@ -1942,8 +1942,8 @@ def static_ifftn( used, or all axes if `s` is also not specified. Repeated indices in axes means inverse transform over that axis is performed multiple times. norm - Optional argument, "backward", "ortho" or "forward". Defaults to be - "backward". + Optional argument, "backward", "ortho" or "forward". + Defaults to be "backward". "backward" indicates no normalization. "ortho" indicates normalization by 1/sqrt(n). "forward" indicates normalization by 1/n. @@ -1987,7 +1987,7 @@ def ifftn( Parameters ---------- x - Input array of complex numbers. + Input array of complex numbers. s sequence of ints, optional @@ -2000,8 +2000,8 @@ def ifftn( used, or all axes if `s` is also not specified. Repeated indices in axes means inverse transform over that axis is performed multiple times. norm - Optional argument, "backward", "ortho" or "forward". Defaults to be - "backward". + Optional argument, "backward", "ortho" or "forward". + Defaults to be "backward". "backward" indicates no normalization. "ortho" indicates normalization by 1/sqrt(n). "forward" indicates normalization by 1/n. diff --git a/ivy/functional/ivy/experimental/layers.py b/ivy/functional/ivy/experimental/layers.py index e6f17d7a73296..0fa38498abdd3 100644 --- a/ivy/functional/ivy/experimental/layers.py +++ b/ivy/functional/ivy/experimental/layers.py @@ -2313,29 +2313,29 @@ def ifftn( Parameters ---------- - x : array_like + x Input array of complex numbers. - s : sequence of ints, optional + s Shape (length of transformed axis) of the output (`s[0]` refers to axis 0, `s[1]` to axis 1, etc.). If given shape is smaller than that of the input, the input is cropped. If larger, input is padded with zeros. If `s` is not given, shape of input along axes specified by axes is used. - axes : sequence of ints, optional + axes Axes over which to compute the IFFT. If not given, last `len(s)` axes are used, or all axes if `s` is also not specified. Repeated indices in axes means inverse transform over that axis is performed multiple times. - norm : {"backward", "ortho", "forward"}, optional + norm Indicates direction of the forward/backward pair of transforms is scaled and with what normalization factor. "backward" indicates no normalization. "ortho" indicates normalization by $\frac{1}{\sqrt{n}}$. "forward" indicates normalization by $\frac{1}{n}$. - out : Optional[ivy.Array], optional + out Optional output array for writing the result to. It must have a shape that the inputs broadcast to. Returns ------- - out : + out The truncated or zero-padded input, transformed along the axes indicated by axes, or by a combination of s or x, as explained in the parameters section above. diff --git a/ivy_tests/test_ivy/test_functional/test_experimental/test_nn/test_layers.py b/ivy_tests/test_ivy/test_functional/test_experimental/test_nn/test_layers.py index b7092a64bdc6a..c9505aa9c3426 100644 --- a/ivy_tests/test_ivy/test_functional/test_experimental/test_nn/test_layers.py +++ b/ivy_tests/test_ivy/test_functional/test_experimental/test_nn/test_layers.py @@ -1101,8 +1101,6 @@ def test_ifftn( fw=backend_fw, on_device=on_device, fn_name=fn_name, - # rtol_=1e-2, - # atol_=1e-2, x=x, s=s, axes=axes,