Tril

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	√
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√
Atlas 200I/500 A2 推理产品	×
Atlas 推理系列产品	×
Atlas 训练系列产品	√

功能说明

• 算子功能：将输入的self张量的最后二维（按shape从左向右数）沿对角线的右上部分置零。参数diagonal可正可负，默认为零，正数表示主对角线向右上方向移动，负数表示主对角线向左下方向移动。

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

  $$\begin{gathered} 对角线及其左下方，即i+d>=j，保留原值：ou{t}_{i,j}=sel{f}_{i,j} \\ 而位于对角线右上方的情况，即i+d<j，置零（不含对角线）：ou{t}_{i,j}=0 \end{gathered}$$

• 示例：

  $self=\left[\begin{array}{ccc}[9& 6& 3]\\ [1& 2& 3]\\ [3& 4& 1]\end{array}\right]$， triu(self, diagonal=0)的结果为： $\left[\begin{array}{ccc}[9& 0& 0]\\ [1& 2& 0]\\ [3& 4& 1]\end{array}\right]$； 调整diagonal的值，triu(self, diagonal=1)结果为： $\left[\begin{array}{ccc}[9& 6& 0]\\ [1& 2& 3]\\ [3& 4& 1]\end{array}\right]$； 调整diagonal为-1，triu(self, diagonal=-1)结果为： $\left[\begin{array}{ccc}[0& 0& 0]\\ [1& 0& 0]\\ [3& 4& 0]\end{array}\right]$。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
self	输入	待进行tril计算的入参，公式中的self。	FLOAT、FLOAT16、DOUBLE、BFLOAT16、INT8、INT16、INT32、INT64、UINT8、UINT16、UINT32、UINT64、BOOL、COMPLEX32、COMPLEX64	ND
diagonal	输入	待进行tril计算的入参，公式中的diagonal。	INT64	ND
out	输出	待进行tril计算的出参，公式中的out_i。	FLOAT、FLOAT16、DOUBLE、BFLOAT16、INT8、INT16、INT32、INT64、UINT8、BOOL	ND

• Atlas 训练系列产品：数据类型支持DOUBLE、FLOAT、FLOAT16、INT16、INT32、INT64、INT8、UINT16、UINT32、UINT64、UINT8、BOOL。
• Atlas A2 训练系列产品/Atlas A2 推理系列产品、Atlas A3 训练系列产品/Atlas A3 推理系列产品：数据类型支持DOUBLE、FLOAT、FLOAT16、INT16、INT32、INT64、INT8、UINT16、UINT32、UINT64、UINT8、BOOL、BFLOAT16。
• Ascend 950PR/Ascend 950DT AI处理器：数据类型支持DOUBLE、FLOAT、FLOAT16、INT16、INT32、INT64、INT8、UINT16、UINT32、UINT64、UINT8、BOOL、BFLOAT16、COMPLEX32、COMPLEX64。

约束说明

无

调用说明

调用方式	调用样例	说明
aclnn调用	test_aclnn_tril	通过aclnnTril接口方式调用Tril算子。