Tensorflow API

来源:互联网 发布:淘宝客买家号 编辑:程序博客网 时间:2024/06/05 06:56

tf.gather_nd

a = tf.constant([[1 , 2, 3, 4, 5, 6],  [10, 1, 2, 3, 4, 0]])o = tf.gather_nd(a, [[0,0], [1,1]])with tf.Session() as s:print s.run(o)# output : [1, 1] 


0 0
原创粉丝点击