开发者
资源
[object Object]

[object Object][object Object]undefined
[object Object]
  • 接口功能:将输入的self张量的最后二维(按shape从左向右数)沿对角线的左下部分置零。参数diagonal可正可负,默认为零,正数表示主对角线向右上方向移动,负数表示主对角线向左下方向移动。

  • 计算公式:下面用i表示遍历倒数第二维元素的序号(i是行索引),用j表示遍历最后一维元素的序号(j是列索引),用d表示diagonal,在(i, j)对应的二维坐标图中,i+d==j表示在对角线上。

    对角线及其右上方,即i+d<=j,保留原值:outi,j=selfi,j而位于对角线左下方的情况,即i+d>j,置零(不含对角线):outi,j=0对角线及其右上方,即i+d<=j,保留原值: out_{i, j} = self_{i, j}\\ 而位于对角线左下方的情况,即i+d>j,置零(不含对角线):out_{i, j} = 0
  • 示例:

    self=[[963][123][341]]self = \begin{bmatrix} [9&6&3] \\ [1&2&3] \\ [3&4&1] \end{bmatrix}

    triu(self, diagonal=0)的结果为:

    [[963][023][001]]\begin{bmatrix} [9&6&3] \\ [0&2&3] \\ [0&0&1] \end{bmatrix}

    调整diagonal的值,triu(self, diagonal=1)结果为:

    [[063][003][000]]\begin{bmatrix} [0&6&3] \\ [0&0&3] \\ [0&0&0] \end{bmatrix}

    调整diagonal为-1,triu(self, diagonal=-1)结果为:

    [[963][123][041]]\begin{bmatrix} [9&6&3] \\ [1&2&3] \\ [0&4&1] \end{bmatrix}

[object Object]
  • aclnnTriu和aclnnInplaceTriu实现相同的功能,使用区别如下,请根据自身实际场景选择合适的算子。
    • aclnnTriu:需新建一个输出张量对象存储计算结果。
    • aclnnInplaceTriu:无需新建输出张量对象,直接在输入张量的内存中存储计算结果。
  • 每个算子分为,必须先调用 “aclnnTriuGetWorkspaceSize” 或者 “aclnnInplaceTriuGetWorkspaceSize” 接口获取入参并根据计算流程计算所需workspace大小,再调用 “aclnnTriu” 或者 “aclnnInplaceTriu” 接口执行计算。
[object Object]
[object Object]
[object Object]
[object Object]
[object Object]
  • 参数说明

    [object Object]
    • Atlas 350 加速卡:COMPLEX32、COMPLEX64仅支持该产品。
    • [object Object]Atlas 推理系列产品[object Object]、[object Object]Atlas 训练系列产品[object Object]:不支持BFLOAT16数据类型。
  • 返回值

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

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

    [object Object]
[object Object]
  • 参数说明

    [object Object]
  • 返回值

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

[object Object]
  • 参数说明

    [object Object]
    • Atlas 350 加速卡:COMPLEX32、COMPLEX64仅支持该产品。
    • [object Object]Atlas 推理系列产品[object Object]、[object Object]Atlas 训练系列产品[object Object]:不支持BFLOAT16数据类型。
  • 返回值

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

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

    [object Object]
[object Object]
  • 参数说明

    [object Object]
  • 返回值

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

[object Object]
  • 确定性计算:
    • aclnnTriu&aclnnInplaceTriu默认确定性实现。
[object Object]

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

[object Object]