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行
相关文章
相关标签/搜索