Method capture

tensorflow/python/framework/function.py:810–818

Adds the given tensor to this graph and returns the captured tensor.

(self, tensor, name=None)

                                             dtypes=dtypes, **kwargs)

  def capture(self, tensor, name=None):
    """Adds the given tensor to this graph and returns the captured tensor."""
    if tensor in self._captured:
      # Captured already.
      return self._captured[tensor]
    elif self._capture_by_value:
      return self._add_tensor_and_parents(tensor)
    else:
      return self._capture_tensor_as_extra_input(tensor, name)

  def _capture_tensor_as_extra_input(self, tensor, name=None):
    # Substitute with a placeholder.