コード例 #1
0
ファイル: math_grad.py プロジェクト: vmichals/tensorflow 
def _BatchFFT3DGrad(_, grad):
    size = math_ops.cast(_FFTSizeForGrad(grad, 3), dtypes.float32)
    return math_ops.batch_ifft3d(grad) * math_ops.complex(size, 0.)
コード例 #2
0
ファイル: math_grad.py プロジェクト: 0ruben/tensorflow 
def _BatchFFT3DGrad(_, grad):
  size = math_ops.cast(_FFTSizeForGrad(grad, 3), dtypes.float32)
  return math_ops.batch_ifft3d(grad) * math_ops.complex(size, 0.)