tf.reduce_mean

JamFF

关注

阅读 76

2023-01-13


import numpy as np
import tensorflow as tf

x = np.array([[1.,2.,3.],[4.,5.,6.]])
sess = tf.Session()

mean1 = sess.run(tf.reduce_mean(x))
mean2 = sess.run(tf.reduce_mean(x, 0))
mean3 = sess.run(tf.reduce_mean(x, 1))


print (x)
print (mean1)
print (mean2)
print (mean3)


sess.close()

[[ 1.  2.  3.]
[ 4. 5. 6.]]
3.5
[ 2.5 3.5 4.5]
[ 2. 5.]


精彩评论(0)

0 0 举报