tf.nn.embedding_lookup函数的用法
tf.nn.embedding_lookup函数的用法主要是选取一个张量里面索引对应的元素。tf.nn.embedding_lookup（params, ids）:params可以是张量也可以是数组等，id就是对应的索引，其他的参数不介绍。例如：ids只有一行：#c = np.random.random()# 随机生成一个10*1的数组#b = tf.nn.embedding_lookup(c, )#查找数组中的序号为1和3的
p=tf.Variable(tf.random_normal())#生成10*1的张量
b = tf.nn.embedding_lookup(p, )#查找张量中的序号为1和3的
with tf.Session() as sess:
sess.run(tf.global_variables_initializer())
print(sess.run(b))
#print(c)
print(sess.run(p))
页:
[1]