FilesLast commitLast update
4 months ago
3 months ago
4 months ago
4 months ago
4 months ago
4 months ago
4 months ago
4 months ago
README

InplaceUpdate

产品支持情况

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

功能说明

  • 算子功能:根据indices将updates中的值更新到x中,实现原地更新操作。

  • 计算公式:

y=x.copy();y[indices[i],...]=v[i,...]y = x.copy();\quad y[\text{indices}[i], ...] = v[i, ...]

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x 输入 被更新的输入张量,支持1D~8D。 FLOAT16、FLOAT32 ND
indices 输入 int32类型的一维向量,指定要更新的索引。 INT32 ND
v 输入 更新值张量,与x同dtype,第0维大小等于indices的长度。 FLOAT16、FLOAT32 ND
y 输出 更新后的张量,shape与x相同,dtype与x相同。 FLOAT16、FLOAT32 ND

约束说明

输入输出仅支持ND格式,x和v的 dtype 必须相同,v.shape[0]必须等于indices的长度,x.shape[1:] 必须等于 v.shape[1:]。当 indices 中存在重复值时,结果为非确定性。

调用说明

调用方式 调用样例 说明
图模式调用 test_geir_inplace_update 参见算子调用完成算子编译和验证。