本文共 690 字，大约阅读时间需要 2 分钟。

PyTorch API 文档

torch.triu 函数说明

torch.triu 是 PyTorch 中一个功能强大的工具，用于返回上三角矩阵。该函数对于数据科学家和工程师在处理矩阵运算时非常有用。

函数定义

torch.triu(input, diagonal=0, out=None) → Tensor

参数说明

功能描述

torch.triu 函数的作用是从输入张量中提取上三角矩阵。上三角矩阵是指矩阵中所有元素的下三角部分（包括对角线）为空。该函数默认返回新的张量，除非你指定了 out 参数，此时函数将直接修改原张量。

示例

a = torch.randn(3, 3)# 返回的结果如下：tensor([[ 0.2309,  0.5207,  2.0049],        [ 0.0000, -1.0680,  0.6602],        [ 0.0000,  0.0000, -0.4573]])

注意事项

• 对角线处理：默认情况下，函数会从对角线开始提取上三角矩阵。如果你想从其他对角线开始，可以通过指定 diagonal 参数来实现。
• 输出形式：如果你希望函数返回新的张量，而不是修改原有张量，可以通过设置 out 参数来实现。

参考文档

PyTorch 官方文档：torch.triu

如果你有任何问题或需要进一步的帮助，请随时联系 PyTorch 社区或查看官方文档。

转载地址：http://mzrfk.baihongyu.com/