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