Trilu

功能

获取一个二维矩阵或一批二维矩阵的上三角或下三角部分,即输入Tensor的shape为[*, N, M],其中*表示零个或多个维度。

输入

input:输入Tensor,数据类型支持float、float16、float64、int64,shape为[*, N, M]。

k(可选):用于指定主对角线的偏移量,若k > 0,则将对主角线向右上方移动;若k < 0,则将主对角线向左下方移动;否则,不移动主对角线。

属性

upper:int,用于指定保留矩阵的上三角还是下三角部分,若为1,保留上三角部分;若为0,保留下三角部分

输出

output:输出Tensor,数据类型和shape与input保持一致。

限制与约束

无。

支持的ONNX版本

Opset v14/v15/v16/v17/v18