| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
| 1 个月前 | ||
| 2 天前 | ||
| 2 天前 | ||
| 13 天前 | ||
| 2 天前 | ||
| 13 天前 | ||
| 9 天前 |
Trilu
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Ascend 950PR/Ascend 950DT | √ |
| Atlas A3 训练系列产品/Atlas A3 推理系列产品 | × |
| Atlas A2 训练系列产品/Atlas A2 推理系列产品 | × |
| Atlas 200I/500 A2 推理产品 | × |
| Atlas 推理系列产品 | × |
| Atlas 训练系列产品 | × |
功能说明
-
算子功能：对输入张量的最后两个维度构成的矩阵，根据
upper属性选择保留上三角或下三角区域的元素，其余元素置零。 -
计算公式：
当upper = 1时（上三角模式）:y[row,col]={x[row,col]if col−row≥diagonal0otherwise\text{当upper = 1时（上三角模式）:} \quad y[row, col] = \begin{cases} x[row, col] & \text{if } col - row \ge diagonal \\ 0 & \text{otherwise} \end{cases}
当upper = 0时（下三角模式）:y[row,col]={x[row,col]if col−row≤diagonal0otherwise\text{当upper = 0时（下三角模式）:} \quad y[row, col] = \begin{cases} x[row, col] & \text{if } col - row \le diagonal \\ 0 & \text{otherwise} \end{cases}
参数说明
| 参数名 | 输入/输出/属性 | 描述 | 数据类型 | 数据格式 |
|---|---|---|---|---|
| x | 输入 | 待进行三角掩码计算的入参，rank至少为2，最后两维为矩阵。 | FLOAT、FLOAT16、DOUBLE、BFLOAT16、INT8、INT16、INT32、INT64、UINT8、UINT16、UINT32、UINT64、BOOL、COMPLEX32、COMPLEX64 | ND |
| k | 输入（可选） | 对角线偏移量，0为主对角线，正数为超对角线，负数为次对角线。默认值为0。 | INT32、INT64 | ND |
| upper | 属性 | 三角区域标志，1表示上三角，0表示下三角。默认值为0。 | INT64 | - |
| y | 输出 | 三角掩码后的输出张量，shape与输入x完全一致。 | 与输入x相同 | ND |
- Ascend 950PR/Ascend 950DT：数据类型支持UINT64、INT64、UINT32、 INT32、UINT16、INT16、UINT8、 INT8、FLOAT16、FLOAT32、DOUBLE、BOOL、BFLOAT16、COMPLEX32、COMPLEX64。
约束说明
- 输入
x必须满足rank >= 2（ND格式），最后两维构成矩阵。 - 不支持0D和1D输入；此类输入在Tiling阶段返回
GRAPH_FAILED。
调用说明
| 调用方式 | 调用样例 | 说明 |
|---|---|---|
| 图模式调用 | test_geir_trilu | 在host侧通过GE IR构图方式调用trilu算子。 |