tf.reduce_mean,求平均值

import tensorflow as tf
import numpy as np

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

#  tf.reduce_mean(
#      input_tensor,
#      axis=None,
#      keepdims=None,
#      name=None,
#      reduction_indices=None,
#      keep_dims=None
#      )

c = np.array([[3.,4], [5.,6], [6.,7]])

step = tf.reduce_mean(c, 1)
with tf.Session() as sess:
    print(sess.run(step))

输出

[3.5 5.5 6.5]

也就是
[3.,4]的平均值是3.5,类推

用例对应的源代码,觉得有帮助可以Star

发布了758 篇原创文章 · 获赞 35 · 访问量 60万+

猜你喜欢

转载自blog.csdn.net/kelsel/article/details/83690118