如何在Pytorch中优化较低的Cholesky参数？

Question

有什么方法可以创建带有正对角线的下三角形的参数，并在Pytorch的优化过程中强制执行此约束？

Answer 1

选中一个torch.potrf。

一个简单的例子：

a = torch.randn(3, 3)
a = torch.mm(a, a.t()) # make symmetric positive definite
l = torch.potrf(a, upper=False)

tri_loss = l.sum()
opt.zero_grad()
tri_loss.backward()
opt.step()