mindscience.sciops
dft
1D discrete cosine transformation on real number on the last axis. |
|
1/2/3D discrete Fourier transformation on complex number. |
|
1D discrete sine transformation on real number on the last axis. |
|
1D inverse discrete cosine transformation on real number on the last axis. |
|
1/2/3D discrete inverse Fourier transformation on complex number. |
|
1D inverse discrete sine transformation on real number on the last axis. |
|
1/2/3D discrete inverse real Fourier transformation on complex number. |
|
1/2/3D discrete real Fourier transformation on real number. |
einsum
Einsum operation using Einstein summation convention. |
evoformer_attention
Performs evoformer attention computation using a custom NPU operator. |
fft
1D complex-to-complex forward FFT transform using Ascend NPU acceleration. |
|
2D complex-to-complex forward FFT transform using Ascend NPU acceleration. |
|
1D complex-to-complex inverse FFT transform using Ascend NPU acceleration. |
|
2D complex-to-complex inverse FFT transform using Ascend NPU acceleration. |
|
1D complex-to-real inverse FFT transform using Ascend NPU acceleration. |
|
2D complex-to-real inverse FFT transform using Ascend NPU acceleration. |
|
1D real-to-complex FFT transform using Ascend NPU acceleration. |
|
2D real-to-complex FFT transform using Ascend NPU acceleration. |
|
1D complex-to-complex forward FFT transform using Ascend NPU acceleration. |
|
2D complex-to-complex forward FFT transform using Ascend NPU acceleration. |
|
N-dimensional complex-to-complex forward FFT transform using Ascend NPU acceleration. |
|
1D complex-to-complex inverse FFT transform using Ascend NPU acceleration. |
|
2D complex-to-complex inverse FFT transform using Ascend NPU acceleration. |
|
N-dimensional complex-to-complex inverse FFT transform using Ascend NPU acceleration. |
|
1D complex-to-real inverse FFT transform using Ascend NPU acceleration. |
|
2D complex-to-real inverse FFT transform using Ascend NPU acceleration. |
|
N-dimensional complex-to-real inverse FFT transform using Ascend NPU acceleration. |
|
1D real-to-complex FFT transform using Ascend NPU acceleration. |
|
2D real-to-complex FFT transform using Ascend NPU acceleration. |
|
N-dimensional real-to-complex FFT transform using Ascend NPU acceleration. |
|
Set cache number of ASD FFT operators to optimize the function call performance. |