[object Object]

[object Object][object Object]undefined[object Object]

• 接口功能：计算KL散度
• 计算公式：

  • 定义loss_pointwise，保存中间结果。

    $$loss\_pointwise_i=\begin{cases} NaN & \text{ if }&logTarget=false \text{ and } target_i <= 0, \\ target_i * \left ( \log{(target_i)}- self_i \right ) & \text{ if }& logTarget=false， \\ \exp^ {target_i} * \left ( target_i- self_i \right ) & \text{ else. } \end{cases}$$

  • out计算公式为：

    $$out=\begin{cases} \bar{loss\_pointwise} & \text{ if }& reduction= 1, \\ \sum loss\_pointwise & \text{ elif }& reduction= 2,\\ \frac{\sum loss\_pointwise}{self.size(0)} & \text{ elif }& reduction= 3,\\ loss\_pointwise & \text{ else. } \end{cases}$$

[object Object]

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

[object Object]

• 参数说明：

  [object Object]
  • [object Object]Atlas 推理系列产品[object Object]、[object Object]Atlas 训练系列产品[object Object]：不支持BFLOAT16。

• 返回值：

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

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

  [object Object]

[object Object]

• 参数说明：

  [object Object]

• 返回值：

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

[object Object]

• 确定性计算：
  • aclnnKlDiv默认确定性实现。

[object Object]

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