火炬 | | 从PyPI进行简单安装: pip install torchkbnufft 关于 torchkbnufft在PyTorch中使用Kaiser-Bessel网格实现了非均匀的快速傅立叶变换[ ]。 该实现完全是在Python中进行的,从而便于在无需编译的情况下以可读代码的灵活部署。 NUFFT函数每个都包装为torch.autograd.Function ,允许通过NUFFT运算符进行反向传播以训练神经网络。 该软件包的主要灵感来自的NUFFT实现。 操作模式和阶段 该软件包具有NUFFT操作模式的三大类:基于表的NUFFT插值,基于稀疏矩阵的NUFFT插值以及具有嵌入Toeplitz的FFT的前向/后向运算符[ ]。 大致而言,计算速度如下: 类型 速度 托普利茨 最快的 桌子 中等的 稀疏矩阵 慢(不推荐) 通常最好从表插值开始,然后尝试其他模式解决问题。