昇腾社区首页
中文
注册

aclnnForeachErf

支持的产品型号

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

接口原型

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

  • aclnnStatus aclnnForeachErfGetWorkspaceSize(const aclTensorList *x, aclTensorList *out, uint64_t *workspaceSize, aclOpExecutor **executor)
  • aclnnStatus aclnnForeachErf(void *workspace, uint64_t workspaceSize, aclOpExecutor *executor, aclrtStream stream)

功能描述

  • 算子功能:

    返回一个和输入张量列表同样形状大小的新张量列表,按元素做误差函数运算(也称之为高斯误差函数,error function or Gauss error function)。

  • 计算公式:

    out=erf(x)=2π0xet2dtout = {\rm erf}(x) = \frac{2}{\sqrt{\pi}} \int_{0}^{x} e^{-t^2} dt

aclnnForeachErfGetWorkspaceSize

  • 参数说明

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

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

    [object Object]

aclnnForeachErf

  • 参数说明

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

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

约束与限制

无。

调用示例

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

[object Object]