mindscience.sciops.fft.asd_rfft

mindscience.sciops.fft.asd_rfft(*args, **kwargs)[源代码]

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

此函数对实数输入张量执行1D实数快速傅里叶变换, 针对Ascend NPU硬件加速进行了优化。

参数:
  • *args - 可变长度参数列表。通常包括:

    • xr (Tensor): 输入实数张量,数据类型为float32。

  • **kwargs - 任意关键字参数。

返回:

Tuple[Tensor, Tensor]。包含以下内容的元组,

  • yr (Tensor),输出复数张量的实部,数据类型为float32。

  • yi (Tensor),输出复数张量的虚部,数据类型为float32。

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

样例:

>>> import mindspore as ms
>>> from mindscience.sciops.fft import asd_rfft
>>> xr = ms.Tensor([[1.0, 2.0, 3.0, 4.0]], ms.float32)
>>> yr, yi = asd_rfft(xr)
>>> print(yr.shape)
(1, 3)
>>> print(yi.shape)
(1, 3)