tf.reduce_mean 函数原型: tf.math.reduce_mean( input_tensor, axis=None, keepdims=None, name=None, reduction_indices=None, keep_dims=None ) 参数: input_tensor:需要求平均值的张量 axis:指定求平均值的轴 keep_dims:是否保存input_tensor的维度不变。