torch.triu函数

>>> 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]])