跳到正文
原文
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