torch.triu()

torch.triu()是一个PyTorch函数，用于返回输入张量的右上角三角部分。其中，“triu”代表“triangular, upper”，表示它会返回一个上三角形矩阵。

该函数可以接受两个参数：输入张量和一个可选的对角线偏移参数。对于输入维度大于等于2的张量，该函数将保留第二维及其以上的所有维度，并且仅在最后两维的右上角形成三角形。

以下是一个简单的示例：

import torch
a = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
b = torch.triu(a)
print(b)


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

在此示例中，我们创建了一个形状为（3，3）的张量“a”。然后我们使用torch.triu()函数返回了“a”的右上角部分。注意，在右上角三角形之外的元素都被填充为零。