mindscience.sciops.fft.ASD_FFT

class mindscience.sciops.fft.ASD_FFT[source]

1D complex-to-complex forward FFT transform using Ascend NPU acceleration.

This operator performs 1D Fast Fourier Transform on complex input tensors, optimized for Ascend NPU hardware acceleration.

Inputs:

  • xr (Tensor): Real part of input complex tensor with data type float32.

  • xi (Tensor): Imaginary part of input complex tensor with data type float32.

Outputs:

  • yr (Tensor): Real part of output complex tensor with data type float32.

  • yi (Tensor): Imaginary part of output complex tensor with data type float32.

Raises

ValueError – If input tensor data type is not float32.

Examples

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