摘要: tf.gather:用一个一维的索引数组,将张量中对应索引的向量提取出来 1 import tensorflow as tf 2 3 a = tf.Variable([[1,2,3,4,5], [6,7,8,9,10], [11,12,13,14,15]]) 4 index_a = tf.Varia 阅读全文
posted @ 2019-11-05 21:59 you-wh 阅读(240) 评论(0) 推荐(0) 编辑
Fork me on GitHub