Method reduce_mean

tensorflow/python/ops/parallel_for/pfor.py:1020–1032

Performs a mean reduction on `x` across pfor iterations. Note that this currently may not work inside a control flow construct. Args: x: an unvectorized Tensor. Returns: A Tensor that has same rank as `x`. The value is the mean of the values of `x` across the pfor ite

(self, x)

      return self._reverse_reduce_concat_map[x]

  def reduce_mean(self, x):
    """Performs a mean reduction on `x` across pfor iterations.

    Note that this currently may not work inside a control flow construct.
    Args:
      x: an unvectorized Tensor.

    Returns:
      A Tensor that has same rank as `x`. The value is the mean of the values
      of `x` across the pfor iterations.
    """
    y = self.reduce_concat(x)
    return math_ops.reduce_mean(y, axis=0)

  def reduce_sum(self, x):
    """Performs a sum reduction on `x` across pfor iterations.