slicemap on TrackedArrays produces Array{TrackedReal}

[baggepinnen]

This is a problem since the code below will not run on the GPU (unless one allows scalar operations which is not ideal)

julia> using SliceMap, Flux

julia> slicemap(norm, Flux.param(randn(2,2,2,2)), dims=(1,2))
2×2 Array{Tracker.TrackedReal{Float64},2}:
 1.91925  2.69252
 1.26406  1.22966