昇腾社区首页
中文
注册
开发者
下载

aclnnForeachAddcdivList

支持的产品型号

  • Atlas A2训练系列产品/Atlas 800I A2推理产品。

接口原型

每个算子分为,必须先调用“aclnnForeachAddcdivListGetWorkspaceSize”接口获取入参并根据计算流程计算所需workspace大小,再调用“aclnnForeachAddcdivList”接口执行计算。

  • [object Object]
  • [object Object]

功能描述

  • 算子功能:

    对多个张量进行逐元素加、乘、除操作,并返回一个新的张量列表。

  • 计算公式:

    outi=x1i+x2ix3i×scalarsiout_i = {x}_{1i}+ \frac{{x}_{2i}}{{x}_{3i}}\times{scalars_i}

aclnnForeachAddcdivListGetWorkspaceSize

  • 参数说明

    • x1(aclTensorList*,计算输入):公式中的[object Object],Device侧的aclTensorList,数据类型支持FLOAT、FLOAT16、BFLOAT16。shape维度不高于8维,支持ND。shape与入参[object Object][object Object][object Object]和出参[object Object]的shape一致。支持
    • x2(aclTensorList*,计算输入):公式中的[object Object],Device侧的aclTensorList,数据类型支持FLOAT、FLOAT16、BFLOAT16。shape维度不高于8维,支持ND。数据类型、数据格式和shape跟[object Object]入参一致。支持
    • x3(aclTensorList*,计算输入):公式中的[object Object],Device侧的aclTensorList,数据类型支持FLOAT、FLOAT16、BFLOAT16。shape维度不高于8维,支持ND。数据类型、数据格式和shape跟[object Object]入参一致。支持
    • scalars(aclTensor*,计算输入):公式中的[object Object],Device侧的aclTensor,数据类型支持FLOAT、FLOAT16、BFLOAT16。shape维度不高于8维,支持ND。数据类型、数据格式和shape跟[object Object]入参一致。支持
    • out(aclTensorList*,计算输出):公式中的[object Object],Device侧的aclTensorList,数据类型支持FLOAT、FLOAT16、BFLOAT16。shape维度不高于8维,支持ND。数据类型、数据格式和shape跟[object Object]入参一致。支持
    • workspaceSize(uint64_t*,出参):返回用户需要在Device侧申请的workspace大小。
    • executor(aclOpExecutor**,出参):返回op执行器,包含了算子计算流程。
  • 返回值

    aclnnStatus:返回状态码,具体参见

    [object Object]

aclnnForeachAddcdivList

  • 参数说明

    • workspace(void*,入参):在Device侧申请的workspace内存地址。
    • workspaceSize(uint64_t,入参):在Device侧申请的workspace大小,由第一段接口aclnnForeachAddcdivListGetWorkspaceSize获取。
    • executor(aclOpExecutor*,入参):op执行器,包含了算子计算流程。
    • stream(aclrtStream,入参):指定执行任务的AscendCL Stream流。
  • 返回值

    aclnnStatus:返回状态码,具体参见

约束与限制

无。

调用示例

示例代码如下,仅供参考,具体编译和执行过程请参考

[object Object]