1 tf.reduce_mean
f.reduce_mean(input_tensor, axis=None, keep_dims=False, name=None, reduction_indices=None)
实际就是直接求平均值。
比如有个矩阵:
x =
[[1, 2, 3]
[4, 5, 6]]
则当:
mean_none = tf.reduce_mean(x)
mean_0 = tf.reduce_mean(x, 0)
mean_1 = tf.reduce_mean(x, 1)
结果为:
mean_none=3.5
mean_0=[2.5, 3.5, 4.5]
mean_1=[2.0, 5.0]
本文详细解析了 TensorFlow 中的 reduce_mean 函数,通过一个矩阵示例展示了如何使用该函数来计算平均值,以及如何指定轴进行操作。reduce_mean 函数在深度学习和机器学习中用于数据预处理和模型训练。
403

被折叠的 条评论
为什么被折叠?



