Make `aggregate_fun` more user friendly

[p-gw]

Automatically broadcast aggregate_fun over the relevant dimensions.
For users this should be more accessibly since they only have to specify mean, median or similar instead of a more difficult function definition like x -> vec(mean(x, dims=2))