コード例 #1
0
ファイル: fft_test.py プロジェクト: Huoxubeiyin/tensorflow
 def _tf_fn(x):
   return spectral_ops.rfft2d(
       x, fft_length=[x.shape[-2].value, x.shape[-1].value])
コード例 #2
0
 def _tf_fn(x):
     return spectral_ops.rfft2d(
         x, fft_length=[x.shape[-2].value, x.shape[-1].value])