tf.reduce_mean

tf.reduce_mean是TensorFlow中的一个函数，用于计算张量的平均值。它可以将张量的每个元素的平均值求出来。

使用方法：

tf.reduce_mean(input_tensor, axis=None, keepdims=False, name=None)

参数解释：

- input_tensor：要计算平均值的张量。
- axis：轴，需要计算平均值的轴。如果为None，则对张量的所有元素求平均值。
- keepdims：是否保留轴维度。如果为True，则结果张量的维度与原张量相同，其他轴上的平均值都是1。
- name：操作的名称。

返回值：计算出的平均值。

例如：

import tensorflow as tf

x = tf.constant([[1, 2], [3, 4]])

mean = tf.reduce_mean(x)  # mean = 2.5

mean = tf.reduce_mean(x, axis=0)  # mean = [2, 3]
mean = tf.reduce_mean(x, axis=1)  # mean = [1.5, 3.5]

mean = tf.reduce_mean(x, keepdims=True)  # mean = [[2.5]]