torch.triu

torch.triu(input, diagonal=0, out=None) → Tensor   返回矩阵上三角部分

参数：

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行对角线的元素；

import torch
a = torch.ones(3, 3)
print(a)
print(torch.triu(a))
print(torch.triu(a, 1))
print(torch.triu(a, -1))
print(1-torch.triu(a))

tensor([[1., 1., 1.],
        [1., 1., 1.],
        [1., 1., 1.]])
tensor([[1., 1., 1.],
        [0., 1., 1.],
        [0., 0., 1.]])
tensor([[0., 1., 1.],
        [0., 0., 1.],
        [0., 0., 0.]])
tensor([[1., 1., 1.],
        [1., 1., 1.],
        [0., 1., 1.]])
tensor([[0., 0., 0.],
        [1., 0., 0.],
        [1., 1., 0.]])

参考：torch.triu(input, diagonal=0, out=None)_danerer的专栏-CSDN博客_torch.triu

最后

以上就是奋斗西牛为你收集整理的torch.triu的全部内容，希望文章能够帮你解决torch.triu所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。