reduce elementwise devisions by 1 in fftfreq and rfftfreq
@@ -896,7 +896,7 @@ def fftfreq(n, d=1.0):
.. seealso:: :func:`numpy.fft.fftfreq`
"""
return cupy.hstack((cupy.arange(0, (n - 1) // 2 + 1, dtype=np.float64),
- cupy.arange(-(n // 2), 0, dtype=np.float64))) / n / d
+ cupy.arange(-(n // 2), 0, dtype=np.float64))) / (n * d)
def rfftfreq(n, d=1.0):
@@ -912,7 +912,7 @@ def rfftfreq(n, d=1.0):
.. seealso:: :func:`numpy.fft.rfftfreq`
"""
- return cupy.arange(0, n // 2 + 1, dtype=np.float64) / n / d
+ return cupy.arange(0, n // 2 + 1, dtype=np.float64) / (n * d)
def fftshift(x, axes=None):
