aclnnOneHot

产品支持情况

[object Object]undefined

功能说明

• 算子功能：对长度为n的输入self， 经过one_hot的计算后得到一个元素数量为n*k的输出out，其中k的值为numClasses。 输出的元素满足下列公式：$$out[i][j]=\left\{ \begin{aligned} onValue,\quad self[i] = j \\ offValue, \quad self[i] \neq j \end{aligned} \right.$$

函数原型

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

aclnnOneHotGetWorkspaceSize

• 参数说明：

[object Object]

• [object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object]：不支持UINT8数据类型。

• 返回值：

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

  [object Object]

aclnnOneHot

• 参数说明：

[object Object]

• 返回值：

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

约束说明

• 确定性计算：
  • aclnnOneHot默认确定性实现。
• [object Object]Atlas 推理系列产品[object Object]、[object Object]Atlas 训练系列产品[object Object]、[object Object]Atlas A2 训练系列产品/Atlas A2 推理系列产品[object Object]、[object Object]Atlas A3 训练系列产品/Atlas A3 推理系列产品[object Object]：
  • 当offValue的数据类型为INT64时，其首元素取值仅支持0或1。
  • 输入self大小为selfSize，输出out大小为outSize，ub大小为ubSize，当axis取值为0，满足下列条件的场景暂不支持：
    • selfSize * 3 < ubSize - 16K < outSize * 3 / 2

调用示例

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