tensorflow函数tf.segment_max，tf.nn.top_k使用小记

import tensorflow as tf

segment_ids=tf.constant([0,0,0,1,1,2])
v=tf.Variable([5.,2.,3.,6.,1.,2.])
m=tf.segment_max(v,segment_ids)
top=tf.nn.top_k(v,k=2,sorted=False)[0]

with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    print(sess.run(v))
    print(sess.run(m))
    print(sess.run(top))