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
网友评论