Merge branch 'master' into torfjelde/product-bijector

TuringLang · Apr 18, 2024 · d6356f7 · d6356f7
2 parents 52af91d + e522155
commit d6356f7
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 1 deletion.
diff --git a/Project.toml b/Project.toml
@@ -1,6 +1,6 @@
 name = "Bijectors"
 uuid = "76274a88-744f-5084-9051-94815aaf08c4"
-version = "0.13.8"
+version = "0.13.9"
 
 [deps]
 ArgCheck = "dce04be8-c92d-5529-be00-80e4d2c0e197"

diff --git a/src/bijectors/simplex.jl b/src/bijectors/simplex.jl
@@ -136,6 +136,12 @@ function logabsdetjac(b::SimplexBijector, x::AbstractVector{T}) where {T}
 
     return -lp
 end
+
+# Needed to avoid falling back to `with_logabsdet_jacobian` for matrix inputs.
+function logabsdetjac(b::SimplexBijector, x::AbstractMatrix{<:Real})
+    return sum(Base.Fix1(logabsdetjac, b), eachcol(x))
+end
+
 function simplex_logabsdetjac_gradient(x::AbstractVector)
     T = eltype(x)
     ϵ = _eps(T)

diff --git a/test/interface.jl b/test/interface.jl
@@ -230,6 +230,7 @@ end
         filldist(Exponential(), 2),
         filldist(Exponential(), 2, 3),
         filldist(filldist(Exponential(), 2), 3),
+        filldist(Dirichlet(ones(2)), 3),
     ]
         x = rand(dist)
         b = bijector(dist)