pytorch triu

PyTorch中的`triu`函数是一个矩阵运算工具，它从对角线及其上方的部分选择元素，返回一个新的张量，保留原始矩阵的三角形部分（上三角）。这个操作通常用于提取稀疏矩阵的非对角线元素，或者创建低秩近似。`triu`函数接受一个张量作为输入，并可以选择指定的k值，表示对角线上k个单位格以外的元素将被过滤掉。

例如：

import torch

# 创建一个二维张量
matrix = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
triu_matrix = torch.triu(matrix)
print(triu_matrix)  # 输出: tensor([[1., 0., 0.],
#                                    [4., 5., 0.],
#                                    [7., 8., 9.]])

如果你想包括对角线，可以设置`k=0`：

triu_with_diagonal = torch.triu(matrix, k=0)
print(triu_with_diagonal)  # 输出: tensor([[1., 2., 3.],
#                                                  [4., 5., 6.],
#                                                  [7., 8., 9.]])