昇腾社区首页
中文
注册

aclnnLogit

产品支持情况

产品 是否支持
[object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object]
[object Object]Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件[object Object]
[object Object]Atlas 200I/500 A2 推理产品[object Object] ×
[object Object]Atlas 推理系列产品 [object Object] ×
[object Object]Atlas 训练系列产品[object Object] ×

功能说明

  • 算子功能:

    该算子是概率到对数几率(log-odds)转换的一个数学运算,常用于概率值的反变换。它可以将一个介于0和1之间的概率值转换为一个实数值。

  • 计算公式:

    yi=ln(zi1zi)y_i = \ln \left( \frac{z_i}{ 1 -z_i} \right)

    其中:

    zi={eps,if xi<epsxi,if epsxi1eps1eps,if xi>1epsz_i = \begin{cases} \text{eps}, & \text{if } x_i < \text{eps} \\ x_i, & \text{if } \text{eps} \leq x_i \leq 1 - \text{eps} \\ 1 - \text{eps}, & \text{if } x_i > 1 - \text{eps} \end{cases}

函数原型

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

  • aclnnStatus aclnnLogitGetWorkspaceSize(const aclTensor *input, double eps, const aclTensor *out, uint64_t *workspaceSize, aclOpExecutor **executor)
  • aclnnStatus aclnnLogit(void *workspace, uint64_t workspaceSize, aclOpExecutor *executor, aclrtStream stream)

aclnnLogitGetWorkspaceSize

  • 参数说明

    • input(aclTensor*,计算输入):输入的张量,公式中的x,Device侧的aclTensor,数据类型支持BFLOAT16、FLOAT16、FLOAT。shape维度0至8维。支持undefinedundefined支持ND。支持空Tensor。
    • eps(double,计算输入):输入input的epsilon限制边界,防止除0错误,公式中的eps,数据类型为double,建议值-1。
    • out(aclTensor*,计算输出):输出张量,公式中的输出y,Device侧的aclTensor,数据类型支持BFLOAT16、FLOAT16、FLOAT。支持undefinedundefined支持ND,输出的数据类型、数据格式和shape与输入input保持一致。
    • workspaceSize(uint64_t*,出参):返回用户需要在Device侧申请的workspace大小。
    • executor(aclOpExecutor**,出参):返回op执行器,包含了算子计算流程。
  • 返回值

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

    [object Object]

aclnnLogit

  • 参数说明

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

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

约束说明

  • input取值范围不在[0, 1],输出nan,取值0时为-inf,取值为1时为inf。
  • eps大于1时,输出nan,取值为1时为inf。

调用示例

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

[object Object]