Private API 
This is the private API reference for Boltz.jl. You know what this means. Don't use these functions!
Boltz.Utils.flatten_spatial Method
julia
flatten_spatial(x::AbstractArray{T, 4})Flattens the first 2 dimensions of x, and permutes the remaining dimensions to (2, 1, 3).
Boltz.Utils.second_dim_mean Method
julia
second_dim_mean(x)Computes the mean of x along dimension 2.
Boltz.Utils.should_type_assert Method
julia
should_type_assert(x)In certain cases, to ensure type-stability we want to add type-asserts. But this won't work for exotic types like ForwardDiff.Dual. We use this function to check if we should add a type-assert for x.