tf.reduce_mean函数用法及有趣区别

sess=tf.Session()

a=np.array([1,2,3,5.])   # 此代码保留为浮点数

a1=np.array([1,2,3,5])   # 此代码保留为整数

c=tf.reduce_mean(a)
d=sess.run(c)
print(a)
print(d)
c1=tf.reduce_mean(a1)
d1=sess.run(c1)
print(a1)
print(d1)

总结：tf.reduce_mean（a，axis）是均值，其中a是输入矩阵，axis是从什么维度求均值。然而，代码运行发现，a为浮点数，返回为浮点数，a为整数，返回为向下取的整数。