Trilu¶
Trilu - 14¶
版本¶
名称: Trilu (GitHub)
域:
main
起始版本:
14
函数:
False
支持级别:
SupportType.COMMON
形状推断:
True
此版本的算子自版本 14 起可用。
摘要¶
给定一个 2-D 矩阵或批量 2-D 矩阵,返回张量的上三角或下三角部分。“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 个输入。
输入 (异构) - T
秩为 2 或更高的输入张量。
k (可选,异构) - tensor(int64)
一个 0-D 张量,包含一个单独的值,表示要排除或包含的主对角线上方或下方的对角线数量。如果未指定,默认值为 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)
)约束输入和输出类型为所有张量类型。