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