From f7f5bda363750fb624448b138dd5871fcece317d Mon Sep 17 00:00:00 2001 From: Arpan Chakraborty Date: Mon, 27 Apr 2026 08:37:26 +0530 Subject: [PATCH] Add docstring for pool_chain function --- src/chains.jl | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/src/chains.jl b/src/chains.jl index 39eeb7186..aa9b12151 100644 --- a/src/chains.jl +++ b/src/chains.jl @@ -851,6 +851,13 @@ function _cat(::Val{3}, c1::Chains, args::Chains...) return Chains(value, missing, c1.name_map, new_info) end +""" + pool_chain(c::Chains) + +Combine all chains in `c` into a single chain by concatenating iterations. + +Returns a new `Chains` object with a single chain containing all iterations from all chains. +""" function pool_chain(c::Chains) data = c.value.data pool_data = reshape(permutedims(data, [1, 3, 2]), :, size(data, 2), 1)