mindscience.sciops.fft.asd_irfft
- mindscience.sciops.fft.asd_irfft(*args, **kwargs)[源代码]
使用Ascend NPU加速的1D复数到实数反向FFT变换。
此函数对复数输入张量执行1D反向实数快速傅里叶变换, 针对Ascend NPU硬件加速进行了优化。
- 参数:
*args - 可变长度参数列表。通常包括:
xr (Tensor): 输入复数张量的实部,数据类型为float32。
xi (Tensor): 输入复数张量的虚部,数据类型为float32。
**kwargs - 任意关键字参数。
- 返回:
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) >>> yr = asd_irfft(xr, xi) >>> print(yr.shape) (1, 4)