tf.nn.embedding_lookup函数

tf.nn.embedding_lookup(params, ids, partition_strategy='mod', name=None, validate_indices=True, max_norm=None)

首先通过一下一个简单的例子来了解一下tf.nn.embedding_lookup()的用法

a = tf.constant([[1,2,3],[4,5,6],[7,8,9],[10,11,12]])

>>>a.eval()

Out[51]:

array([[ 1, 2, 3],

[ 4, 5, 6],

[ 7, 8, 9],

[10, 11, 12]], dtype=int32)

   

tf.nn.embedding_lookup(a,[0,1]).eval()

>>>Out[52]:

array([[1, 2, 3],

[4, 5, 6]], dtype=int32)

   

tf.nn.embedding_lookup(a,[[0,1],[1,2]]).eval()

>>>Out[53]:

array([[[1, 2, 3],

[4, 5, 6]],

[[4, 5, 6],

[7, 8, 9]]], dtype=int32)

posted @   simple_wxl  阅读(316)  评论(0编辑  收藏  举报
编辑推荐:
· Linux glibc自带哈希表的用例及性能测试
· 深入理解 Mybatis 分库分表执行原理
· 如何打造一个高并发系统?
· .NET Core GC压缩(compact_phase)底层原理浅谈
· 现代计算机视觉入门之:什么是图片特征编码
阅读排行:
· 手把手教你在本地部署DeepSeek R1,搭建web-ui ,建议收藏!
· Spring AI + Ollama 实现 deepseek-r1 的API服务和调用
· 数据库服务器 SQL Server 版本升级公告
· C#/.NET/.NET Core技术前沿周刊 | 第 23 期(2025年1.20-1.26)
· 程序员常用高效实用工具推荐,办公效率提升利器!
点击右上角即可分享
微信分享提示