Private API

This is the private API reference for Boltz.jl. You know what this means. Don't use these functions!

Boltz.Utils.fast_chunk Method

fast_chunk(x::AbstractArray, ::Val{n}, ::Val{dim})

Type-stable and faster version of MLUtils.chunk.

source

Boltz.Utils.flatten_spatial Method

flatten_spatial(x::AbstractArray{T, 4})

Flattens the first 2 dimensions of x, and permutes the remaining dimensions to (2, 1, 3).

source

Boltz.Utils.second_dim_mean Method

second_dim_mean(x)

Computes the mean of x along dimension 2.

source

Boltz.Utils.should_type_assert Method

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.

source