1. 程式人生 > >tf.reduce_mean,求平均值

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