torch.triu

`torch.triu(input, diagonal=0)` 是一个 PyTorch 函数，用于获取矩阵的上三角部分。

参数 `input` 是一个张量，可以是任意形状的矩阵。参数 `diagonal` 是一个整数，表示从主对角线开始向上偏移的位置。默认值为 0，表示从主对角线开始。

函数返回一个新的张量，其中上三角部分被保留，其余部分被填充为0。例如，对于一个 3x3 的矩阵，`torch.triu()` 函数将返回一个 3x3 的矩阵，其中只有上三角部分是原始矩阵的值，其余部分为0。

示例代码：

import torch

x = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
y = torch.triu(x)

print(y)

输出结果为：

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