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

最新推荐文章于 2025-06-09 07:50:41 发布

danerli

最新推荐文章于 2025-06-09 07:50:41 发布

阅读量1.8w

点赞数 13

CC 4.0 BY-SA版权

分类专栏： PyTorch Python

本文链接：https://blog.youkuaiyun.com/danerer/article/details/82932891

Python 同时被 2 个专栏收录

15 篇文章

订阅专栏

PyTorch

8 篇文章

订阅专栏

本文详细介绍了 PyTorch 中的 triu 函数，该函数用于从输入矩阵中提取上三角部分，并将剩余部分设为零。文章通过实例展示了如何使用 diagonal 参数来控制上三角部分的具体范围。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

torch.triu(input, diagonal=0, out=None) → Tensor
返回矩阵上三角部分，其余部分定义为0。
Parameters:

input (Tensor) – the input tensor
diagonal (int, optional) – the diagonal to consider
out (Tensor, optional) – the output tensor
如果diagonal为空，输入矩阵保留主对角线与主对角线以上的元素；
如果diagonal为正数n，输入矩阵保留主对角线与主对角线以上除去n行的元素；
如果diagonal为负数-n，输入矩阵保留主对角线与主对角线以上与主对角线下方h行对角线的元素；

>>> a = torch.randn(3, 3)
>>> a
tensor([[ 0.2309,  0.5207,  2.0049],
        [ 0.2072, -1.0680,  0.6602],
        [ 0.3480, -0.5211, -0.4573]])
>>> torch.triu(a)
tensor([[ 0.2309,  0.5207,  2.0049],
        [ 0.0000, -1.0680,  0.6602],
        [ 0.0000,  0.0000, -0.4573]])
>>> torch.triu(a, diagonal=1)
tensor([[ 0.0000,  0.5207,  2.0049],
        [ 0.0000,  0.0000,  0.6602],
        [ 0.0000,  0.0000,  0.0000]])
>>> torch.triu(a, diagonal=-1)
tensor([[ 0.2309,  0.5207,  2.0049],
        [ 0.2072, -1.0680,  0.6602],
        [ 0.0000, -0.5211, -0.4573]])