matrix - 我们如何检查矩阵是否为 PSD 是否为 PyTorch？

Question

有一个poste在 Python 中检查矩阵是否为 PSD。我想知道我们如何在 PyTorch 中检查它？有这个功能吗？

Answer 1

尚未为此找到 PyTorch 函数，但您应该能够轻松确定它，与您链接的帖子类似，通过检查矩阵是否对称以及所有特征值是否为非负:

def is_psd(mat):
    return bool((mat == mat.T).all() and (torch.eig(mat)[0][:,0]>=0).all())
#Test:
is_psd(torch.randn(2,2))