tensorflow的reduce_sum()方法

2017-12-11  本文已影响0人  theo_NI

reduce_sum() 就是求和,由于求和的对象是tensor,所以是沿着tensor的某些维度求和。reduction_indices是指沿tensor的哪些维度求和。(可省略不写)

import tensorflow as tf
sess=tf.Session()
s=[[1,1,1],
   [2,2,2]]
a=tf.reduce_sum(s)
print(sess.run(a))   # 9
b=tf.reduce_sum(s,0)
print(sess.run(b))   #  [3 3 3]
c=tf.reduce_sum(s,1)
print(sess.run(c))   #  [3 6]
d=tf.reduce_sum(s,1,keep_dims=True)# [[3],[6]]
print(sess.run(d))
f=tf.reduce_sum(s,[0,1])
print(sess.run(f))    #  9
上一篇 下一篇

猜你喜欢

热点阅读