python reduce_sum使用記録

576 ワード

import tensorflow as tf
data=[[1,1,2],[1,1,1]]

#    
defaultReduceSume=tf.reduce_sum(data)
#    
hangLieReduceSumBy=tf.reduce_sum(data, [0, 1])


#    
hangReduceSum=tf.reduce_sum(data,1)
#    
lieReduceSum=tf.reduce_sum(data,0)

#        ,true     
hangZReduceSum=tf.reduce_sum(data, 1, keep_dims=True)



with tf.Session() as sess:
    print(sess.run(defaultReduceSume))
    print(sess.run(hangLieReduceSumBy))
    print(sess.run(hangReduceSum))
    print(sess.run(lieReduceSum))
    print(sess.run(hangZReduceSum))

印刷結果:7 7[43][22 3][[4][3]]