Trilu

Trilu - 14

版本

• 名称: Trilu (GitHub)

• 域: main

• 自版本: 14

• 函数: False

• 支持级别: SupportType.COMMON

• 形状推断: True

此版本的运算符自版本 14起可用。

摘要

给定一个二维矩阵或二维矩阵的批次，返回张量（张量）的上三角或下三角部分。属性“upper”决定保留上三角还是下三角部分。如果设置为 true，则保留上三角矩阵。否则保留下三角矩阵。属性“upper”的默认值为 true。Trilu 接受一个形状为 [*, N, M] 的输入张量，其中 * 是零个或多个批次维度。上三角部分包含给定对角线 (k) 上及以上的元素。下三角部分包含对角线及以下的元素。矩阵中的所有其他元素都设置为零。如果 k = 0，则保留主对角线上及以上/以下的三角部分。如果 upper 设置为 true，则正 k 会保留上三角矩阵，不包括主对角线和 (k-1) 个对角线。负 k 值保留主对角线和 |k| 个对角线。如果 upper 设置为 false，则正 k 会保留下三角矩阵，包括主对角线和 k 个对角线。负 k 值不包括主对角线和 (|k|-1) 个对角线。

属性

• upper - INT (默认值为 '1')

  布尔值。指示保留矩阵的上三角部分还是下三角部分。默认值为 true。

输入

1 到 2 个输入之间。

• input (异构) - T

  等级为 2 或更高的输入张量。

• k (可选，异构) - tensor(int64)

  一个包含单个值的 0 维张量，对应于要排除或包含的主对角线以上或以下的对角线条数。如果没有指定，则默认值为 0。

输出

• output (异构) - T

  与输入张量类型和形状相同的输出张量。

类型约束

• T in ( tensor(bfloat16), tensor(bool), tensor(complex128), tensor(complex64), tensor(double), tensor(float), tensor(float16), tensor(int16), tensor(int32), tensor(int64), tensor(int8), tensor(string), tensor(uint16), tensor(uint32), tensor(uint64), tensor(uint8) )

  将输入和输出类型约束为所有张量类型。