tf.nn.embedding_lookup()函数

1、tf.nn.embedding_lookup() tf.nn.embedding_lookup函数的用法主要是选取一个张量里面索引对应的元素。tf.nn.embedding_lookup(tensor, id):tensor就是输入张量,id就是张量对应的索引。ide 2、代码 import tensorflow as tf import numpy as np input_ids = tf
相关文章
相关标签/搜索