From 62ebc1ddf9fe563c1499ecaf78463535fdfb07ec Mon Sep 17 00:00:00 2001
From: Misha Mikhasenko <mikhail.mikhasenko@gmail.com>
Date: Wed, 16 Jul 2025 22:36:43 +0200
Subject: [PATCH] RationalQuadraticSpline for batch dimension

---
 src/bijectors/rational_quadratic_spline.jl | 4 ++++
 1 file changed, 4 insertions(+)

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 ###
 ##########################