Exemplo n.º 1
0
def _IFFT3DGrad(_, grad):
  rsize = math_ops.cast(
      1. / math_ops.cast(_FFTSizeForGrad(grad, 3), grad.dtype.real_dtype),
      grad.dtype)
  return spectral_ops.fft3d(grad) * rsize
Exemplo n.º 2
0
def _IFFT3DGrad(_, grad):
    rsize = 1. / math_ops.cast(_FFTSizeForGrad(grad, 3), dtypes.float32)
    return spectral_ops.fft3d(grad) * math_ops.complex(rsize, 0.)
Exemplo n.º 3
0
def _IFFT3DGrad(_, grad):
    rsize = math_ops.cast(
        1. / math_ops.cast(_FFTSizeForGrad(grad, 3), grad.dtype.real_dtype),
        grad.dtype)
    return spectral_ops.fft3d(grad) * rsize
Exemplo n.º 4
0
def _IFFT3DGrad(_, grad):
  rsize = 1. / math_ops.cast(_FFTSizeForGrad(grad, 3), dtypes.float32)
  return spectral_ops.fft3d(grad) * math_ops.complex(rsize, 0.)