ruijiaosun / fft_dtensor Goto Github PK
View Code? Open in Web Editor NEWA function calculating multi-dimensional Fourier Transform for DTensors. DTensor is an extension to TensorFlow for synchronous distributed computing (https://www.tensorflow.org/guide/dtensor_overview)