tf.nn.embedding_lookup()用法

welcome to my blog tf.nn.embedding_lookup(params, ids)的作用:ids裏面是params中的索引, 執行該函數將從params中選出ids索引對應的元素 具體地,創建一個形狀爲(6,4)的數組, 通過tf.nn.embedding_lookup(params, ids)取出該二維數組的第1行和第3行
相關文章
相關標籤/搜索