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
相關文章
相關標籤/搜索