torch.fft
Note
若API“是否支持”为“是”,“限制与说明”为“-”,说明此API和原生API支持度保持一致。
使用以下接口,需要配置以下信息:
source {CANN包安装路径}/nnal/atb/set_env.sh
source {CANN包安装路径}/nnal/asdsip/set_env.sh
| API名称 | 是否支持 | 限制与说明 |
|---|---|---|
| torch.fft.rfftn | 是 | 支持fp32 支持1-8维。2维维度为(batch, n), 1维维度为(n) 1. batch维度:[1, 8, 16, 24, 32, 64] 2. n维度: - 2n2^n,n为23以内 - 2, 3, 5, 7任意相乘,例如: 2225779,但结果需在1000000以内 - 200以内质数任意相乘,结果需在100000以内 取值范围[-100, 100] |
| torch.fft.hfft | 是 | - |
| torch.fft.ihfft | 是 | - |
| torch.fft.hfft2 | 是 | - |
| torch.fft.ihfft2 | 否 | - |
| torch.fft.hfftn | 是 | - |
| torch.fft.ihfftn | 否 | - |
| torch.fft.fftfreq | 否 | - |
| torch.fft.rfftfreq | 否 | - |
| torch.fft.fftshift | 否 | - |
| torch.fft.ifftshift | 否 | - |