ef2e13c4719c6a9263748c5428ee070893763c5b,pyro/distributions/hmm.py,,_logmatmulexp,#Any#Any#,36

Before Change

    Numerically stable version of ``(x.log() @ y.log()).exp()``.
    
    x_shift = x.max(-1, keepdim=True)[0]
    y_shift = y.max(-2, keepdim=True)[0]
    xy = torch.matmul((x - x_shift).exp(), (y - y_shift).exp()).log()
    return xy + x_shift + y_shift

After Change

    
    Numerically stable version of ``(x.log() @ y.log()).exp()``.
    
    finfo = torch.finfo(x.dtype)  // avoid nan due to -inf - -inf
    x_shift = x.max(-1, keepdim=True).values.clamp(min=finfo.min)
    y_shift = y.max(-2, keepdim=True).values.clamp(min=finfo.min)
    xy = safe_log(torch.matmul((x - x_shift).exp(), (y - y_shift).exp()))
    return xy + x_shift + y_shift

Instances