Skip to content

Commit

Permalink
Backport rfft-like input dtype fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
honno committed Oct 9, 2023
1 parent b2731b2 commit 1528ff3
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions src/array_api_stubs/_2022_12/fft.py
Original file line number Diff line number Diff line change
Expand Up @@ -279,15 +279,15 @@ def irfft(
norm: Literal["backward", "ortho", "forward"] = "backward",
) -> array:
"""
Computes the one-dimensional inverse of ``rfft`` for complex-valued input.
Computes the one-dimensional inverse of ``rfft`` for real-valued input.
.. note::
Applying the one-dimensional inverse discrete Fourier transform for real-valued input to the output of this function must return the original (i.e., non-transformed) input array within numerical accuracy (i.e., ``irfft(rfft(x)) == x``), provided that the transform and inverse transform are performed with the same arguments (axis and normalization mode) and consistent length.
Parameters
----------
x: array
input array. Should have a complex-valued floating-point data type.
input array. Must have a real-valued floating-point data type.
n: int
length of the transformed axis of the **output**. If
Expand Down Expand Up @@ -388,15 +388,15 @@ def irfftn(
norm: Literal["backward", "ortho", "forward"] = "backward",
) -> array:
"""
Computes the n-dimensional inverse of ``rfftn`` for complex-valued input.
Computes the n-dimensional inverse of ``rfftn`` for real-valued input.
.. note::
Applying the n-dimensional inverse discrete Fourier transform for real-valued input to the output of this function must return the original (i.e., non-transformed) input array within numerical accuracy (i.e., ``irfftn(rfftn(x)) == x``), provided that the transform and inverse transform are performed with the same arguments (axes and normalization mode) and consistent sizes.
Parameters
----------
x: array
input array. Should have a complex-valued floating-point data type.
input array. Must have a real-valued floating-point data type.
s: Sequence[int]
size of each transformed axis of the **output**. ``n=s[i]`` is also the number of input points used along the axis (dimension) ``i``, except for the last axis, where ``n=s[-1]//2+1`` points of the input are used. If
Expand Down

0 comments on commit 1528ff3

Please sign in to comment.