Merge pull request #152 from biaslab/gradlogpartition_exponential

add getgradlogpartition Exponential
ReactiveBayes · Dec 15, 2023 · decb514 · decb514
2 parents 65f5e6b + 9ad7b0a
commit decb514
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/distributions/exponential.jl b/src/distributions/exponential.jl
@@ -44,6 +44,11 @@ getlogpartition(::NaturalParametersSpace, ::Type{Exponential}) = (η) -> begin
     return -log(-η₁)
 end
 
+getgradlogpartition(::NaturalParametersSpace, ::Type{Exponential}) = (η) -> begin
+    (η₁,) = unpack_parameters(Exponential, η)
+    return SA[-1/η₁]
+end
+
 getfisherinformation(::NaturalParametersSpace, ::Type{Exponential}) = (η) -> begin
     (η₁,) = unpack_parameters(Exponential, η)
     SA[inv(η₁^2);;]