[tensorflow] tf.gather使用方法

最新推荐文章于 2022-12-07 23:23:36 发布

weixin_30929295

最新推荐文章于 2022-12-07 23:23:36 发布

阅读量253

点赞数

原文链接：http://www.cnblogs.com/estragon/p/9809154.html

版权

tf.gather：用一个一维的索引数组，将张量中对应索引的向量提取出来

import tensorflow as tf 

a = tf.Variable([[1,2,3,4,5], [6,7,8,9,10], [11,12,13,14,15]])
index_a = tf.Variable([0,2])

b = tf.Variable([1,2,3,4,5,6,7,8,9,10])
index_b = tf.Variable([2,4,6,8])

with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())
    print(sess.run(tf.gather(a, index_a)))
    print(sess.run(tf.gather(b, index_b)))

#  [[ 1  2  3  4  5]
#   [11 12 13 14 15]]

#  [3 5 7 9]

转载于:https://www.cnblogs.com/estragon/p/9809154.html