CANN/ops-nn LpLoss算子
LpLoss【免费下载链接】ops-nn本项目是CANN提供的神经网络类计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-nn产品支持情况产品是否支持Ascend 950PR/Ascend 950DT√Atlas A3 训练系列产品/Atlas A3 推理系列产品√Atlas A2 训练系列产品/Atlas A2 推理系列产品√Atlas 200I/500 A2 推理产品×Atlas 推理系列产品√Atlas 训练系列产品√功能说明算子功能计算输入self和目标target中每个元素之间的平均绝对误差Mean Absolute Error简称MAE。reduction指定要应用到输出的缩减支持 none、mean、sum。none 表示不应用缩减mean 表示输出的总和将除以输出中的元素数sum 表示输出将被求和。计算公式当reduction为none时$$ \ell(x, y) L {l_1,\dots,l_N}^\top, \quad l_n \left| x_n - y_n \right|, $$其中$x$是self$y$是target$N$是batch的大小。如果reduction不是none, 那么$$ \ell(x, y) \begin{cases} \operatorname{mean}(L), \text{if reduction} \text{mean;}\ \operatorname{sum}(L), \text{if reduction} \text{sum.} \end{cases} $$参数说明self (aclTensor*计算输入)公式中的输入selfDevice侧的aclTensor。数据类型与target的数据类型满足数据类型推导规则参见互推导关系。shape支持0-8维shape需要与target满足broadcast规则。支持非连续的Tensor数据格式支持ND。Atlas A2 训练系列产品/Atlas A2 推理系列产品 、 Ascend 950PR/Ascend 950DT 、 Atlas A3 训练系列产品/Atlas A3 推理系列产品 数据类型支持BFLOAT16、FLOAT16、FLOAT32、INT64。target (aclTensor*计算输入)公式中的输入targetDevice侧的aclTensor。数据类型与self的数据类型满足数据类型推导规则参见互推导关系。shape支持0-8维shape需要与self满足broadcast规则。支持非连续的Tensor数据格式支持ND。Atlas A2 训练系列产品/Atlas A2 推理系列产品 、 Ascend 950PR/Ascend 950DT 、 Atlas A3 训练系列产品/Atlas A3 推理系列产品 数据类型支持BFLOAT16、FLOAT16、FLOAT32、INT64。reduction (int64_t计算输入为属性)用于指定要应用到输出的缩减公式中的输入公式中的reductionHost侧的整型。支持 0(none) | 1(mean) | 2(sum)。none 表示不应缩减mean 表示输出的总和将除以输出中的元素数sum 表示输出将被求和。out (aclTensor*计算输出)公式中的outDevice侧的aclTensor。且数据类型需要是self与target推导之后可转换的数据类型参见互转换关系。支持非连续的Tensor数据格式支持ND。当reduction的值为0时out的shape与self和target broadcast后的shape一致当reduction的值不为0时out的shape支持0维。Atlas A2 训练系列产品/Atlas A2 推理系列产品 、 Ascend 950PR/Ascend 950DT 、 Atlas A3 训练系列产品/Atlas A3 推理系列产品 数据类型支持BFLOAT16、FLOAT16、FLOAT32、INT64、COMPLEX64、COMPLEX128。约束说明确定性计算aclnnL1Loss默认确定性实现。LpLoss中p为计算loss的参数只支持p1aclnn接口名称为aclnnL1Loss。调用说明调用方式样例代码说明aclnn接口test_aclnn_l1_loss.cpp通过aclnnL1Loss接口方式调用LpLoss算子。【免费下载链接】ops-nn本项目是CANN提供的神经网络类计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-nn创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/2603556.html
如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!