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