Python TFE_Py_ForwardAccumulatorJVP示例

示例#1

文件： forwardprop.py 项目： Chiranjeevi731/Machine-Learning

 def _fetch_jvp(tensor):
     if hasattr(tensor, "handle"):
         tensor = ops.convert_to_tensor(tensor.handle)
     result = pywrap_tfe.TFE_Py_ForwardAccumulatorJVP(
         self._accumulator, tensor)
     if result is None and unconnected_gradients == UnconnectedGradients.ZERO:
         return array_ops.zeros_like(tensor)
     return result

示例#2

 def _fetch_jvp(tensor):
     if hasattr(tensor, "handle"):
         unwrapped_tensor = ops.convert_to_tensor(tensor.handle)
     else:
         if isinstance(tensor, np_arrays.ndarray):
             unwrapped_tensor = tensor.data
         else:
             unwrapped_tensor = tensor
     result = pywrap_tfe.TFE_Py_ForwardAccumulatorJVP(
         self._accumulator, unwrapped_tensor)
     if result is None and unconnected_gradients == UnconnectedGradients.ZERO:
         result = array_ops.zeros_like(tensor)
     if result is not None and isinstance(tensor, np_arrays.ndarray):
         return np_arrays.tensor_to_ndarray(result)
     return result