Esempio n. 1
0
 def _tf_fn(x):
   return spectral_ops.rfft2d(
       x, fft_length=[x.shape[-2].value, x.shape[-1].value])
Esempio n. 2
0
 def _tf_fn(x):
     return spectral_ops.rfft2d(
         x, fft_length=[x.shape[-2].value, x.shape[-1].value])