single.py 文件源码

python
阅读 23 收藏 0 点赞 0 评论 0

项目:qpth 作者: locuslab 项目源码 文件源码
def factor_kkt(U_S, R, d):
    """ Factor the U22 block that we can only do after we know D. """
    nineq = R.size(0)
    U_S[-nineq:, -nineq:] = torch.potrf(R + torch.diag(1 / d.cpu()).type_as(d))
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号