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

EmbeddingHashTableApplyAdamW

产品支持情况

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

功能说明

  • 算子功能:查询hash表是否存在key,如果存在,则更新其value,如果不存在,则插入key。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
table_handle 输入 输入hash表handle句柄,里面包含了hash表的表头地址等。 INT64 ND
keys 输入 插入key序列。 INT64 ND
m 输入 AdamW算法中修正的一阶矩阵。 FLOAT16、FLOAT ND
v 输入 AdamW算法中修正的二阶矩阵。 FLOAT16、FLOAT ND
beta1_power 输入 前置beta1的次方结果。 FLOAT16、FLOAT ND
beta2_power 输入 前置beta2的次方结果。 FLOAT16、FLOAT ND
lr 输入 学习率。 FLOAT16、FLOAT ND
weight_decay 输入 权重衰减系数。 FLOAT16、FLOAT ND
beta1 输入 用于计算一阶矩阵的系数。 FLOAT16、FLOAT ND
beta2 输入 用于计算二阶矩阵的系数。 FLOAT16、FLOAT ND
epsilon 输入 除0防护,提高数值稳定性。 FLOAT16、FLOAT ND
grad 输入 输入梯度。 FLOAT16、FLOAT ND
max_grad_norm 输入 AMSGrad算法中用于限制norm梯度最大值。 FLOAT16、FLOAT ND
embedding_dim 输入属性 hash表桶深度。 INT64 -
bucket_size 输入属性 hash表桶数量。 INT64 -
amsgrad 输入属性 判断是否使用AMSGrad算法。 BOOL -
maximize 输入属性 判断是否最大化参数。 BOOL -
m 输出 AdamW算法,修正后的一阶矩阵。 FLOAT16、FLOAT ND
v 输出 AdamW算法,修正后的二阶矩阵。 FLOAT16、FLOAT ND
beta1_power 输出 更新后的beta1的次方结果。 FLOAT16、FLOAT ND
beta2_power 输出 更新后的beta2的次方结果。 FLOAT16、FLOAT ND
max_grad_norm 输出 更新后的梯度参数。 FLOAT16、FLOAT ND

约束说明

调用说明

调用方式 样例代码 说明