python - 如何在 Pytorch 中创建上三角矩阵？

Question

简单的问题，但是在 Pytorch 中是否有一种本地方法可以从现有矩阵创建上三角矩阵？我正在考虑使用 mask ，但即使这样也需要创建上三角矩阵。

Answer 1

import torch
upper_tri = torch.ones(rol, col).triu()

例如:

>> mat = torch.ones(3, 3).triu()
>> print(mat)
tensor([[1., 1., 1.],
        [0., 1., 1.],
        [0., 0., 1.]])