aclnnRenorm&aclnnInplaceRenorm
产品支持情况
功能说明
接口功能:返回一个张量,其中输入张量self沿维度dim的每个子张量都经过归一化,使得子张量的p范数低于maxNorm值。
计算公式:
其中: 为dim确定的某维度张量切片:
举例:
[object Object]
函数原型
aclnnRenorm和aclnnInplaceRenorm实现相同的功能,使用区别如下,请根据自身实际场景选择合适的算子。
- aclnnRenorm:需新建一个输出张量对象存储计算结果。
- aclnnInplaceRenorm:无需新建输出张量对象,直接在输入张量的内存中存储计算结果。
每个算子分为,必须先调用“aclnnRenormGetWorkspaceSize”或者“aclnnInplaceRenormGetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器,再调用“aclnnRenorm”或者“aclnnInplaceRenorm”接口执行计算。
[object Object][object Object][object Object][object Object]
aclnnRenormGetWorkspaceSize
参数说明:
[object Object]- [object Object]Atlas 训练系列产品[object Object]:参数
[object Object]、[object Object]的数据类型不支持BFLOAT16。
- [object Object]Atlas 训练系列产品[object Object]:参数
返回值:
第一段接口完成入参校验,出现以下场景时报错:
[object Object]
aclnnRenorm
aclnnInplaceRenormGetWorkspaceSize
参数说明:
[object Object]- [object Object]Atlas 训练系列产品[object Object]:参数
[object Object]的数据类型不支持BFLOAT16。
- [object Object]Atlas 训练系列产品[object Object]:参数
返回值:
第一段接口完成入参校验,出现以下场景时报错:
[object Object]
aclnnInplaceRenorm
约束说明
- 确定性计算:
- aclnnRenorm默认确定性实现。
- aclnnInplaceRenorm默认确定性实现。
调用示例
aclnnRenorm示例代码:
[object Object]aclnnInplaceRenorm示例代码:
[object Object]