mindscience.sciops
离散傅里叶变换
在最后一个轴上对实数进行 1D 离散余弦变换。 |
|
1/2/3D 离散复数傅里叶变换。 |
|
在最后一个轴上对实数进行 1D 离散正弦变换。 |
|
在最后一个轴上对实数进行 1D 离散余弦逆变换。 |
|
1/2/3D 离散复数傅里叶逆变换。 |
|
在最后一个轴上对实数进行 1D 离散正弦逆变换。 |
|
1/2/3D 离散实数傅里叶逆变换。 |
|
1/2/3D 离散实数傅里叶变换。 |
爱因斯坦求和
使用爱因斯坦求和约定(Einsum)进行张量计算的操作。 |
evoformer注意力
使用自定义NPU算子执行evoformer注意力计算。 |
快速傅里叶变换
使用Ascend NPU加速的1D复数到复数前向FFT变换。 |
|
使用Ascend NPU加速的2D复数到复数前向FFT变换。 |
|
使用Ascend NPU加速的1D复数到复数反向FFT变换。 |
|
使用Ascend NPU加速的2D复数到复数反向FFT变换。 |
|
使用Ascend NPU加速的1D复数到实数反向FFT变换。 |
|
使用Ascend NPU加速的2D复数到实数反向FFT变换。 |
|
使用Ascend NPU加速的1D实数到复数FFT变换。 |
|
使用Ascend NPU加速的2D实数到复数FFT变换。 |
|
使用Ascend NPU加速的1D复数到复数前向FFT变换。 |
|
使用Ascend NPU加速的2D复数到复数前向FFT变换。 |
|
使用Ascend NPU加速的N维复数到复数前向FFT变换。 |
|
使用Ascend NPU加速的1D复数到复数反向FFT变换。 |
|
使用Ascend NPU加速的2D复数到复数反向FFT变换。 |
|
使用Ascend NPU加速的N维复数到复数反向FFT变换。 |
|
使用Ascend NPU加速的1D复数到实数反向FFT变换。 |
|
使用Ascend NPU加速的2D复数到实数反向FFT变换。 |
|
使用Ascend NPU加速的N维复数到实数反向FFT变换。 |
|
使用Ascend NPU加速的1D实数到复数FFT变换。 |
|
使用Ascend NPU加速的2D实数到复数FFT变换。 |
|
使用Ascend NPU加速的N维实数到复数FFT变换。 |
|
设置ASD FFT算子的缓存数量以优化函数调用性能。 |