triu

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

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