文件最后提交记录最后更新时间
10 天前
10 天前
10 天前
10 天前
5 个月前
9 天前
README.md

InitEmbeddingHashTable

产品支持情况

产品 是否支持
Ascend 950PR/Ascend 950DT
Atlas A3 训练系列产品/Atlas A3 推理系列产品
Atlas A2 训练系列产品/Atlas A2 推理系列产品
Atlas 200I/500 A2 推理产品
Atlas 推理系列产品
Atlas 训练系列产品

功能说明

  • 算子功能:初始化hash表。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
table_handle 输入 输入hash表handle句柄,里面包含了hash表的表头地址等。 INT64 ND
sampled_values 输入 外部生成随机数,用于填充value值,initializer_mode为random时生效。 FLOAT ND
bucket_size 输入属性 hash表桶数量。 INT64 ND
embedding_dim 输入属性 hash表桶深度。 INT64 ND
initializer_mode 输入属性 控制初始化的hash表中填充的默认值,random填充sampled_values里面的值,const填充constant_value属性值。 STRING ND
constant_value 输入属性 hash表的初始填充值,initializer_mode为const时生效。 FLOAT ND

约束说明

调用说明

调用方式 样例代码 说明