Exemplo n.º 1
0
 def _tf_fn(x):
     return signal.rfft3d(x,
                          fft_length=[
                              x.shape[-3].value, x.shape[-2].value,
                              x.shape[-1].value
                          ])
Exemplo n.º 2
0
 def _tf_fn(x):
   return signal.rfft3d(
       x,
       fft_length=[
           x.shape[-3].value // 2, x.shape[-2].value, x.shape[-1].value * 2
       ])
Exemplo n.º 3
0
 def _tf_fn(x):
     return signal.rfft3d(
         x, fft_length=[x.shape[-3] // 2, x.shape[-2], x.shape[-1] * 2])