Unfold a nested Julia array #Julia #array

deepflatten.jl

# For a detailed explanation, please see here:
# https://discourse.julialang.org/t/how-do-you-unfold-a-nested-julia-array/2243/7?u=singularitti
function deepflatten(arr::Vector{<: Vector})
    dim = [1]

    function recursiveflatten(arr, dim)
        if arr isa Vector{<: Vector}
            recursiveflatten(collect(Iterators.flatten(arr)),
                pushfirst!(dim, length(arr) / prod(dim)))
        else
            arr, pushfirst!(dim, length(arr) / prod(dim))
        end
    end

    flattened, dim = recursiveflatten(arr, dim)
    reshape(flattened, dim[1:end - 1]...)
end