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