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 -