Pytorch mask：上三角和下三角

上三角 triu

Pytorch上三角和下三角的调用与numpy是相同的。

np.triu(np.ones((5,5)),k=0) # k控制对角线开始的位置
Out[25]: 
array([[1., 1., 1., 1., 1.],
       [0., 1., 1., 1., 1.],
       [0., 0., 1., 1., 1.],
       [0., 0., 0., 1., 1.],
       [0., 0., 0., 0., 1.]])

构建一个上三角mask

torch.triu(torch.ones(5,5),diagonal=0)
Out[17]: 
tensor([[1., 1., 1., 1., 1.],
        [0., 1., 1., 1., 1.],
        [0., 0., 1., 1., 1.],
        [0., 0., 0., 1., 1.],
        [0., 0., 0., 0., 1.]])

dianonal控制上三角的对角线开始位置

torch.triu(torch.ones(5,5),diagonal=1)
Out[20]: 
tensor([[0., 1., 1., 1., 1.],
        [0., 0., 1., 1., 1.],
        [0., 0., 0., 1., 1.],
        [0., 0., 0., 0., 1.],
        [0., 0., 0., 0., 0.]])

下三角 tril

torch.tril(torch.ones(5,5),diagonal=0)
Out[21]: 
tensor([[1., 0., 0., 0., 0.],
        [1., 1., 0., 0., 0.],
        [1., 1., 1., 0., 0.],
        [1., 1., 1., 1., 0.],
        [1., 1., 1., 1., 1.]])

torch.tril(torch.ones(5,5),diagonal=1)
Out[22]: 
tensor([[1., 1., 0., 0., 0.],
        [1., 1., 1., 0., 0.],
        [1., 1., 1., 1., 0.],
        [1., 1., 1., 1., 1.],
        [1., 1., 1., 1., 1.]])

最后

以上就是娇气小海豚为你收集整理的Pytorch mask：上三角和下三角的全部内容，希望文章能够帮你解决Pytorch mask：上三角和下三角所遇到的程序开发问题。

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