pytorch torch.triu

`torch.triu(input, diagonal=0)` 是 PyTorch 中的一个函数，其作用是返回一个张量的上三角矩阵。其中，`input` 是需要进行操作的张量，`diagonal` 是指定对角线的位置，具体含义如下：

- `diagonal=0` 表示不偏移对角线，即返回原始张量的上三角部分；
- `diagonal>0` 表示对角线上移，即返回原始张量对角线以上 `diagonal` 行的上三角部分；
- `diagonal<0` 表示对角线下移，即返回原始张量对角线以下 `diagonal` 列的上三角部分。

例如，对于一个 3x3 的张量 `a`，`torch.triu(a)` 返回的就是其上三角部分，即：

tensor([[1, 2, 3],
        [0, 5, 6],
        [0, 0, 9]])

如果我们指定 `diagonal=1`，则返回的是对角线以上 1 行的上三角部分：

tensor([[0, 2, 3],
        [0, 0, 6],
        [0, 0, 0]])