search

JuliaML / MLUtils.jl

Utilities and abstractions for Machine Learning tasks
MIT License
107 stars 20 forks source link

extend `flatten` with a dims argument #136

Open CarloLucibello opened 1 year ago

CarloLucibello commented 1 year ago

so that flatten(x, dims=1:2) flattens the first two dimensions and so on.