pytorch中，可以获取张量上三角部分内容的函数

在 PyTorch 中，可以使用 `triu()` 函数来获取张量的上三角部分，该函数返回一个新的张量，其中包含输入张量的上三角部分。可以使用 `k` 参数来控制获取的上三角部分相对于主对角线的偏移量，例如，`k=0` 时，获取的是主对角线及其以上的部分，`k=1` 时，获取的是主对角线以下一行及其以上的部分，以此类推。

以下是一个示例代码：

import torch

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

输出结果为：

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

其中，`diagonal` 参数指定了主对角线的位置，`diagonal=0` 表示主对角线在第一行第一列上。函数返回的张量 `y` 中，上三角部分包含了输入张量 `x` 的上三角部分，下三角部分则全部填充为0。