Exemplo n.º 1
0
def _BatchFFT2DGrad(_, grad):
    size = math_ops.cast(_FFTSizeForGrad(grad, 2), dtypes.float32)
    return math_ops.batch_ifft2d(grad) * math_ops.complex(size, 0.)
Exemplo n.º 2
0
def _BatchFFT2DGrad(_, grad):
  size = math_ops.cast(_FFTSizeForGrad(grad, 2), dtypes.float32)
  return math_ops.batch_ifft2d(grad) * math_ops.complex(size, 0.)