diff --git a/src/bijectors/rational_quadratic_spline.jl b/src/bijectors/rational_quadratic_spline.jl index 1ad8acef..77c54fd1 100644 --- a/src/bijectors/rational_quadratic_spline.jl +++ b/src/bijectors/rational_quadratic_spline.jl @@ -177,6 +177,10 @@ function transform(b::RationalQuadraticSpline{<:AbstractMatrix}, x::AbstractVect ] end +function transform(b::RationalQuadraticSpline{<:AbstractMatrix}, x::AbstractMatrix) + return mapreduce(b, hcat, eachcol(x)) +end + ########################## ### Inverse evaluation ### ##########################