TensorFlow Blog· TensorFlow Blog·· 2023-08-25AI 评分31
TensorFlow 分布式快速傅里叶变换(Distributed FFT)功能解析
Distributed Fast Fourier Transform in TensorFlow
AI 导读
summary_zh: TensorFlow 通过 DTensor API 原生支持分布式 FFT,用户只需将分片张量传入现有 FFT 操作(如 tf.signal.fft2d)即可,输出同样为分片张量。
来源:TensorFlow Blog · blog.tensorflow.org