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為整數，返回為向下取的整數。