mindscience.sciops

离散傅里叶变换

mindscience.sciops.dft.DCT

在最后一个轴上对实数进行 1D 离散余弦变换。

mindscience.sciops.dft.DFTn

1/2/3D 离散复数傅里叶变换。

mindscience.sciops.dft.DST

在最后一个轴上对实数进行 1D 离散正弦变换。

mindscience.sciops.dft.IDCT

在最后一个轴上对实数进行 1D 离散余弦逆变换。

mindscience.sciops.dft.IDFTn

1/2/3D 离散复数傅里叶逆变换。

mindscience.sciops.dft.IDST

在最后一个轴上对实数进行 1D 离散正弦逆变换。

mindscience.sciops.dft.IRDFTn

1/2/3D 离散实数傅里叶逆变换。

mindscience.sciops.dft.RDFTn

1/2/3D 离散实数傅里叶变换。

爱因斯坦求和

mindscience.sciops.einsum.Einsum

使用爱因斯坦求和约定(Einsum)进行张量计算的操作。

evoformer注意力

mindscience.sciops.evoformer_attention.evo_attention

使用自定义NPU算子执行evoformer注意力计算。

快速傅里叶变换

mindscience.sciops.fft.ASD_FFT

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

mindscience.sciops.fft.ASD_FFT2D

使用Ascend NPU加速的2D复数到复数前向FFT变换。

mindscience.sciops.fft.ASD_IFFT

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

mindscience.sciops.fft.ASD_IFFT2D

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

mindscience.sciops.fft.ASD_IRFFT

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

mindscience.sciops.fft.ASD_IRFFT2D

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

mindscience.sciops.fft.ASD_RFFT

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

mindscience.sciops.fft.ASD_RFFT2D

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

mindscience.sciops.fft.asd_fft

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

mindscience.sciops.fft.asd_fft2d

使用Ascend NPU加速的2D复数到复数前向FFT变换。

mindscience.sciops.fft.asd_fftn

使用Ascend NPU加速的N维复数到复数前向FFT变换。

mindscience.sciops.fft.asd_ifft

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

mindscience.sciops.fft.asd_ifft2d

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

mindscience.sciops.fft.asd_ifftn

使用Ascend NPU加速的N维复数到复数反向FFT变换。

mindscience.sciops.fft.asd_irfft

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

mindscience.sciops.fft.asd_irfft2d

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

mindscience.sciops.fft.asd_irfftn

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

mindscience.sciops.fft.asd_rfft

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

mindscience.sciops.fft.asd_rfft2d

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

mindscience.sciops.fft.asd_rfftn

使用Ascend NPU加速的N维实数到复数FFT变换。

mindscience.sciops.fft.set_fft_cache_size

设置ASD FFT算子的缓存数量以优化函数调用性能。