mindscience.sciops.fft.ASD_IRFFT

class mindscience.sciops.fft.ASD_IRFFT[源代码]

使用Ascend NPU加速的1D复数到实数反向FFT变换。

此算子对复数输入张量执行1D反向实数快速傅里叶变换，针对Ascend NPU硬件加速进行了优化。

输入：

xr (Tensor) - 输入复数张量的实部，数据类型为float32。
xi (Tensor) - 输入复数张量的虚部，数据类型为float32。

输出：

yr (Tensor) - 输出实数张量，数据类型为float32。

异常：

ValueError - 如果输入张量数据类型不是float32。

样例：

>>> import mindspore as ms
>>> from mindscience.sciops.fft import ASD_IRFFT
>>> xr = ms.Tensor([[1.0, 2.0, 3.0]], ms.float32)
>>> xi = ms.Tensor([[0.0, 0.0, 0.0]], ms.float32)
>>> asd_irfft = ASD_IRFFT()
>>> yr = asd_irfft(xr, xi)
>>> print(yr.shape)
(1, 4)