Skip to content

Commit 62ebc1d

Browse files
authored
RationalQuadraticSpline for batch dimension
1 parent 82d4cdf commit 62ebc1d

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/bijectors/rational_quadratic_spline.jl

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -177,6 +177,10 @@ function transform(b::RationalQuadraticSpline{<:AbstractMatrix}, x::AbstractVect
177177
]
178178
end
179179

180+
function transform(b::RationalQuadraticSpline{<:AbstractMatrix}, x::AbstractMatrix)
181+
return mapreduce(b, hcat, eachcol(x))
182+
end
183+
180184
##########################
181185
### Inverse evaluation ###
182186
##########################

0 commit comments

Comments
 (0)