Lerp

产品支持情况

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

功能说明

  • 算子功能:根据给定的权重,在起始和结束Tensor之间进行线性插值,返回插值后的Tensor。

  • 计算公式:

 out i= start i+ weight i×( end i− start i)\text { out }_i=\text { start }_i+\text { weight }_i \times\left(\text { end }_i-\text { start }_i\right)

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
self 输入 待进行lerp计算的入参,公式中的start_i。 FLOAT、FLOAT16、BFLOAT16 ND
end 输入 待进行lerp计算的入参,公式中的end_i。 FLOAT、FLOAT16、BFLOAT16 ND
weight 输入 待进行lerp计算的入参,公式中的weight_i。 FLOAT、FLOAT16、BFLOAT16 ND
out 输出 待进行lerp计算的出参,公式中的out_i。 FLOAT、FLOAT16、BFLOAT16 ND

约束说明

end、weight的dtype需要与self一致。

调用说明

调用方式 调用样例 说明
aclnn调用 test_aclnn_lerp 通过aclnnLerp接口方式调用Lerp算子。