コード例 #1
0
ファイル: spectral_grad.py プロジェクト: AnishShah/tensorflow 
def _FFT3DGrad(_, grad):
  size = math_ops.cast(_FFTSizeForGrad(grad, 3), grad.dtype)
  return spectral_ops.ifft3d(grad) * size
コード例 #2
0
def _FFT3DGrad(_, grad):
    size = math_ops.cast(_FFTSizeForGrad(grad, 3), dtypes.float32)
    return spectral_ops.ifft3d(grad) * math_ops.complex(size, 0.)
コード例 #3
0
ファイル: spectral_grad.py プロジェクト: neuroph12/CNNDDDD 
def _FFT3DGrad(_, grad):
    size = math_ops.cast(_FFTSizeForGrad(grad, 3), grad.dtype)
    return spectral_ops.ifft3d(grad) * size
コード例 #4
0
def _FFT3DGrad(_, grad):
  size = math_ops.cast(_FFTSizeForGrad(grad, 3), dtypes.float32)
  return spectral_ops.ifft3d(grad) * math_ops.complex(size, 0.)