Skip to content

Commit bc29c40

Browse files
committed
Merge branch 'cr-dirichlet' of github.com:JuliaStats/Distributions.jl into cr-dirichlet
2 parents 1d79fec + ab01122 commit bc29c40

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/multivariate/dirichlet.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -391,8 +391,8 @@ function ChainRulesCore.rrule(::Type{DT}, alpha::AbstractVector{T}; check_args::
391391
d = DT(alpha; check_args=check_args)
392392
function dirichlet_pullback(d_dir)
393393
d_dir = ChainRulesCore.unthunk(d_dir)
394-
∂l = d_dir.lmnB * (SpecialFunctions.digamma.(alpha) .- SpecialFunctions.digamma.(d.alpha0))
395-
return (ChainRulesCore.NoTangent(), d_dir.alpha .+ d_dir.alpha0 .+ ∂l)
394+
dalpha = d_dir.alpha .+ d_dir.alpha0 .+ d_dir.lmnB .* (SpecialFunctions.digamma.(alpha) .- SpecialFunctions.digamma.(d.alpha0))
395+
return ChainRulesCore.NoTangent(), dalpha
396396
end
397397
return d, dirichlet_pullback
398398
end

0 commit comments

Comments
 (0)