如何在 Pytorch 中創建上三角矩陣？

Question

簡單的問題，但是有沒有一種本地方法可以從 Pytorch 中的現有矩陣創建上三角矩陣？ 我正在考慮使用蒙版，但即使這樣也需要創建上三角矩陣。

Answer 1

import torch

l = torch.tril(torch.ones(row, column))

這將返回大小為(row, column)的矩陣的下三角部分。

Answer 2

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