triu

paddle. triu ( x, diagonal=0, name=None )

返回输入矩阵 input 的上三角部分,其余部分被设为 0。 矩形的上三角部分被定义为对角线上和上方的元素。

参数

  • x (Tensor):输入 Tensor x,数据类型支持 float32, float64, int32, int64, complex64, complex128

  • diagonal (int,可选):指定的对角线,默认值为 0。如果 diagonal = 0,表示主对角线;如果 diagonal 是正数,表示主对角线之上的对角线;如果 diagonal 是负数,表示主对角线之下的对角线。

  • name (str,可选) - 具体用法请参见 Name,一般无需设置,默认值为 None。

返回

Tensor,数据类型与输入 input 数据类型一致。

代码示例

COPY-FROM: paddle.triu