aclnnNLLLoss

产品支持情况

[object Object]undefined

功能说明

• 接口功能：计算负对数似然损失值。

• 计算公式：

  当[object Object]为[object Object]时，

  $$\ell(x, y) = L = \{l_1,\dots,l_N\}^\top, \quad l_n = - w_{y_n} x_{n,y_n}, \quad w_{c} = \text{weight}[c] \cdot \mathbb{1}\{c \not= \text{ignoreIndex}\},$$

  其中$x$是self, $y$是target, $w$是weight，$N$是batch的大小. 如果[object Object]不是[object Object], 那么：

  $$\ell(x, y) = \begin{cases} \sum_{n=1}^N \frac{1}{\sum_{n=1}^N w_{y_n}} l_n, & \text{if reduction} = \text{`mean';}\\ \sum_{n=1}^N l_n, & \text{if reduction} = \text{`sum'.} \end{cases}$$ $$totalWeight = \sum_{n=1}^N w_{y_n}$$

函数原型

每个算子分为，必须先调用“aclnnNLLLossGetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器，再调用“aclnnNLLLoss”接口执行计算。

aclnnNLLLossGetWorkspaceSize

• 参数说明

  [object Object]

• 返回值

  aclnnStatus: 返回状态码，具体参见。

  第一段接口完成入参校验，出现以下场景时报错：

  [object Object]

aclnnNLLLoss

• 参数说明

  [object Object]

• 返回值

  aclnnStatus：返回状态码，具体参见。

约束说明

• 确定性计算：
  • aclnnNLLLoss默认非确定性实现，支持通过aclrtCtxSetSysParamOpt开启确定性。

调用示例

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