Minor tweak

TuringLang · Jul 21, 2019 · 74ba08c · 74ba08c
1 parent 3224ee5
commit 74ba08c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/Bijectors.jl b/src/Bijectors.jl
@@ -339,7 +339,7 @@ function logpdf_with_trans(d::PDMatDistribution, X::AbstractMatrix{<:Real}, tran
     lp = logpdf(d, X)
     if transform && isfinite(lp)
         U = cholesky(X).U
-        lp += sum(i->(dim(d) - i + 2) * log(U[i, i]), 1:dim(d))
+        lp += sum((dim(d) .- (1:dim(d)) .+ 2) .* log.(view(U, diagind(U))))
         lp += dim(d) * log(2)
     end
     return lp