aclnnNLLLoss2d

产品支持情况

[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}$$

函数原型

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

aclnnNLLLoss2dGetWorkspaceSize

• 参数说明

  [object Object]

• 返回值

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

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

  [object Object]

aclnnNLLLoss2d

• 参数说明

  [object Object]

• 返回值

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

约束说明

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

调用示例

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