Skip to content

Commit 475a934

Browse files
Update src/multivariate/dirichlet.jl
Co-authored-by: David Widmann <devmotion@users.noreply.github.com>
1 parent e702017 commit 475a934

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/multivariate/dirichlet.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -418,7 +418,7 @@ function ChainRulesCore.rrule(::typeof(_logpdf), d::Dirichlet, x::AbstractVector
418418
function Dirichlet_logpdf_pullback(dy)
419419
∂alpha = xlogy.(dy, x)
420420
∂l = -dy
421-
∂x = dy * (d.alpha .-1) ./ x
421+
∂x = dy .* (d.alpha .-1) ./ x
422422
∂alpha0 = 0.0
423423
if !isfinite(y)
424424
∂alpha .= NaN

0 commit comments

Comments
 (0)