昇腾社区首页
中文
注册
开发者
下载

aclnnCtcLoss

产品支持情况

[object Object]undefined

功能说明

  • 接口功能:计算连接时序分类损失值。

  • 计算公式: 定义ykty_{k}^{t}表示在时刻tt时真实字符为kk的概率。(一般地,ykty_{k}^{t}是经过softmax之后的输出矩阵中的一个元素)。将字符集LL^{'}可以构成的所有序列的集合称为LTL^{'T},将LTL^{'T}中的任意一个序列称为路径,并标记为ππππ的分布为公式(1):

    p(πx)=t=1Tyπtt,πLT.(1)p(π|x)=\prod_{t=1}^{T}y^{t}_{π_{t}} , \forall π \in L'^{T}. \tag{1}

    定义多对一(many to one)映射B: LTLTL^{'T} \to L^{\leq T},通过映射B计算得到lLTl \in L^{\leq T}的条件概率,等于对应于ll的所有可能路径的概率之和,公式(2):

    p(lx)=πB1(l)p(πx).(2)p(l|x)=\sum_{π \in B^{-1}(l)}p(π|x).\tag{2}

    将找到使p(lx)p(l|x)值最大的ll的路径的任务称为解码,公式(3):

    h(x)=lLTarg max p(lx).(3)h(x)=^{arg \ max}_{l \in L^{ \leq T}} \ p(l|x).\tag{3}

    当zeroInfinity为True时

    h(x)={0,h(x)==Inf or h(x)==Infh(x), else h(x)=\begin{cases}0,&h(x) == Inf \text{ or } h(x) == -Inf \\h(x),&\text { else }\end{cases}

函数原型

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

[object Object]
[object Object]

aclnnCtcLossGetWorkspaceSize

  • 参数说明:

    [object Object]
    • [object Object]Atlas 推理系列产品[object Object]、[object Object]Atlas 训练系列产品[object Object]:数据类型不支持FLOAT16、BFLOAT16。
    • logAlphaOut:
      • [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]:shape为(N,T,(2max(targetLengths)+8)/88N, T, (2*max(targetLengths)+8)/8*8)。
  • 返回值:

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

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

    [object Object]

aclnnCtcLoss

  • 参数说明:

    [object Object]
  • 返回值:

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

约束说明

  • 值域限制说明:

    • [object Object]的值域要求为[0,C1][0, C - 1]且不包括blank对应的数值,其中CC代表[object Object]中的最后一维,即类别数。
    • [object Object]的值域要求为[1,T][1, T],其中TT代表[object Object]中的第0维,代表输入长度。
    • [object Object]的值域要求为大于等于1。
    • [object Object]中的元素要求小于等于[object Object]中对应的元素。

    若不满足前三条值域约束,CPU/GPU可能存在越界行为,导致negLogLikelihoodOut和logAlphaOut的计算结果可能与CPU/GPU存在差异。若不满足第四条值域约束,logAlphaOut在对应batch上的计算结果与CPU/GPU存在差异。

  • 确定性计算:

    • aclnnCtcLoss默认非确定性实现,支持通过aclrtCtxSetSysParamOpt开启确定性。

调用示例

  • [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]:示例代码如下,仅供参考,具体编译和执行过程请参考

    [object Object]