예제 #1
0
 def _tf_fn(x):
   return spectral_ops.irfft2d(
       x, fft_length=[x.shape[-2].value, 2 * (x.shape[-1].value - 1)])
예제 #2
0
 def _tf_fn(x):
     return spectral_ops.irfft2d(
         x, fft_length=[x.shape[-2].value, 2 * (x.shape[-1].value - 1)])