tf.reduce_mean 降维求平均值

Aliases:

• tf.math.reduce_mean
• tf.reduce_mean

define

Defined in tensorflow/python/ops/math_ops.py.

Computes the mean of elements across dimensions of a tensor. (deprecated arguments)

SOME ARGUMENTS ARE DEPRECATED. They will be removed in a future version. Instructions for updating: keep_dims is deprecated, use keepdims instead

Reduces input_tensor along the dimensions given in axis. Unless keepdims is true, the rank of the tensor is reduced by 1 for each entry in axis. If keepdims is true, the reduced dimensions are retained with length 1. 降维求均值

If axis is None, all dimensions are reduced, and a tensor with a single element is returned.如果没有指定轴，所有轴都降维求均值。如2维变为一个标量。

example

x = tf.constant([[1., 1.], [2., 2.]])
tf.reduce_mean(x)  # 1.5
tf.reduce_mean(x, 0)  # [1.5, 1.5]
tf.reduce_mean(x, 1)  # [1.,  2.]

Args:

• input_tensor: The tensor to reduce. Should have numeric type. axis:
• The dimensions to reduce. If None (the default), reduces all
  dimensions. Must be in the range [-rank(input_tensor),
  rank(input_tensor)). keepdims: If true, retains reduced dimensions
  with length 1.
• name: A name for the operation (optional).
• reduction_indices: The old (deprecated) name for axis. keep_dims:
  Deprecated alias for keepdims.