The fallback result of `batch` over `AbstractArray{<:Number}`

[findmyway]

julia> x = rand(2,3,4);

julia> batch(x) |> size
(24,)

It seems batch flatten the array here. I think it would make more sense to keep the array as it is here?

batch(x::AbstractArray{<:Number}) = x

Or at least keep the original size?

[CarloLucibello]

I think this should error, unless x is a vector