Fix _logmatmulexp comment (#3105)

pyro-ppl · Jun 8, 2022 · f4b13ec · f4b13ec
1 parent 9022252
commit f4b13ec
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyro/distributions/hmm.py b/pyro/distributions/hmm.py
@@ -48,7 +48,7 @@ def _linear_integrate(init, trans, shift):
 
 def _logmatmulexp(x, y):
     """
-    Numerically stable version of ``(x.log() @ y.log()).exp()``.
+    Numerically stable version of ``(x.exp() @ y.exp()).log()``.
     """
     finfo = torch.finfo(x.dtype)  # avoid nan due to -inf - -inf
     x_shift = x.detach().max(-1, keepdim=True).values.clamp_(min=finfo.min)