CANN Ascend C Math API样例

CANN Ascend C Math API样例 Math API样例介绍概述本样例集介绍了Math API不同特性的典型用法给出了对应的端到端实现。样例列表目录名称功能描述支持的产品acosh本样例演示了基于Acosh高阶API的算子实现。样例按元素做双曲反余弦函数计算Ascend 950PR/Ascend 950DTAtlas A3 训练系列产品/Atlas A3 推理系列产品Atlas A2 训练系列产品/Atlas A2 推理系列产品axpy_half_float本样例基于Axpy实现源操作数src中每个元素与标量求积后和目的操作数dst中的对应元素相加的功能。Axpy接口的源操作数和目的操作数的数据类型只能取三种组合(half, half)、(float, float)、(half, float)。本样例中输入tensor和标量的数据类型为half输出tensor数据类型为float。本样例通过Ascend C编程语言实现了Axpy算子使用内核调用符来完成算子核函数在NPU侧运行验证的基础流程给出了对应的端到端实现Ascend 950PR/Ascend 950DTAtlas A3 训练系列产品/Atlas A3 推理系列产品Atlas A2 训练系列产品/Atlas A2 推理系列产品bitwiseand本样例演示了基于BitwiseAnd高阶API的算子实现。样例逐比特对两个输入进行与操作Ascend 950PR/Ascend 950DTceil本样例演示了基于Ceil高阶API的算子实现。样例获取大于或等于x的最小的整数值即向正无穷取整操作Ascend 950PR/Ascend 950DTAtlas A3 训练系列产品/Atlas A3 推理系列产品Atlas A2 训练系列产品/Atlas A2 推理系列产品clamp本样例演示了基于Clamp高阶API的算子实现。将输入中非nan且超出[min, max]范围的值剪裁至max或min若minmax则所有非nan值均置为maxAscend 950PR/Ascend 950DTclampmin本样例演示了基于ClampMin高阶API的算子实现。样例将srcTensor中小于scalar的数替换为scalar大于等于scalar的数保持不变作为dstTensor输出Ascend 950PR/Ascend 950DTAtlas A3 训练系列产品/Atlas A3 推理系列产品Atlas A2 训练系列产品/Atlas A2 推理系列产品cumsum本样例介绍了调用CumSum高阶API实现cumsum单算子用于对输入张量按行或列进行累加和操作Ascend 950PR/Ascend 950DTAtlas A3 训练系列产品/Atlas A3 推理系列产品Atlas A2 训练系列产品/Atlas A2 推理系列产品erf本样例演示了基于Erf高阶API的算子实现。样例按元素做误差函数计算Ascend 950PR/Ascend 950DTAtlas A3 训练系列产品/Atlas A3 推理系列产品Atlas A2 训练系列产品/Atlas A2 推理系列产品exp本样例演示了基于Exp高阶API的算子实现。样例按元素取自然指数用户可以选择是否使用泰勒展开公式进行计算Ascend 950PR/Ascend 950DTAtlas A3 训练系列产品/Atlas A3 推理系列产品Atlas A2 训练系列产品/Atlas A2 推理系列产品fma本样例演示了基于Fma高阶API的算子实现。样例按元素计算两个输入相乘后与第三个输入相加的结果Ascend 950PR/Ascend 950DTfmod本样例演示了基于Fmod高阶API的算子实现。样例按元素计算两个浮点数ab相除后的余数Ascend 950PR/Ascend 950DTAtlas A3 训练系列产品/Atlas A3 推理系列产品Atlas A2 训练系列产品/Atlas A2 推理系列产品frac本样例演示了基于Frac高阶API的算子实现。样例按元素做取小数计算Ascend 950PR/Ascend 950DTAtlas A3 训练系列产品/Atlas A3 推理系列产品Atlas A2 训练系列产品/Atlas A2 推理系列产品isfinite本样例演示了基于IsFinite高阶API的算子实现。样例按元素判断输入的浮点数是否非NAN、非INF输出结果为浮点数或者布尔值Ascend 950PR/Ascend 950DTlog本样例演示了基于Log高阶API的算子实现。样例按元素以e、2、10为底做对数运算Ascend 950PR/Ascend 950DTAtlas A3 训练系列产品/Atlas A3 推理系列产品Atlas A2 训练系列产品/Atlas A2 推理系列产品logicaland本样例演示了基于LogicalAnd高阶API的算子实现。样例按元素进行与操作输入数据类型不是bool时零被视为False非零数据被视为TrueAscend 950PR/Ascend 950DTpower本样例演示了基于Power高阶API的算子实现。样例实现按元素做幂运算功能支持三种功能指数和底数分别为张量对张量、张量对标量、标量对张量的幂运算Ascend 950PR/Ascend 950DTAtlas A3 训练系列产品/Atlas A3 推理系列产品Atlas A2 训练系列产品/Atlas A2 推理系列产品rint本样例演示了基于Rint高阶API的算子实现。样例获取与输入数据最接近的整数若存在两个相同接近的整数则获取其中的偶数Ascend 950PR/Ascend 950DTwhere本样例演示了基于Where高阶API的算子实现。样例根据指定的条件从两个源操作数中选择元素生成目标操作数。两个源操作数均可以是LocalTensor或标量Ascend 950PR/Ascend 950DTxor本样例演示了基于Xor高阶API的算子实现。样例按元素执行Xor运算Ascend 950PR/Ascend 950DTAtlas A3 训练系列产品/Atlas A3 推理系列产品Atlas A2 训练系列产品/Atlas A2 推理系列产品创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考