aclnnForeachMulScalar
产品支持情况
功能说明
接口功能:对输入张量列表的每个张量与张量scalar执行相乘运算。
计算公式:
函数原型
每个算子分为,必须先调用“aclnnForeachMulScalarGetWorkspaceSize”接口获取入参并根据计算流程计算所需workspace大小,再调用“aclnnForeachMulScalar”接口执行计算。
[object Object]
[object Object]
aclnnForeachMulScalarGetWorkspaceSize
参数说明:
[object Object][object Object]Atlas A2 训练系列产品/Atlas A2 推理系列产品[object Object]、[object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object]:
参数
[object Object]的数据类型与入参[object Object]的数据类型具有一定对应关系:- 当
[object Object]的数据类型为FLOAT32、FLOAT16、INT32时,数据类型与[object Object]的数据类型保持一致。 - 当
[object Object]的数据类型为BFLOAT16时,数据类型支持FLOAT32。
- 当
返回值:
第一段接口完成入参校验,出现以下场景时报错:
[object Object]
aclnnForeachMulScalar
约束说明
- 确定性计算:
- aclnnForeachMulScalar默认确定性实现。
调用示例
[object Object]