Fix broadcasting bug on Julia 1.6 (#74)

Author: adrhill

README.md

@@ -12,7 +12,7 @@ Explainable AI in Julia using [Flux.jl](https://fluxml.ai).
 This package implements interpretability methods and visualizations for neural networks, similar to [Captum][captum-repo] and [Zennit][zennit-repo] for PyTorch and [iNNvestigate][innvestigate-repo] for Keras models. 
 
 ## Installation 
-To install this package and its dependencies, open the Julia REPL and run 
+This package supports Julia ≥1.6. To install it, open the Julia REPL and run 
 ```julia-repl
 julia> ]add ExplainableAI
 ```

src/input_augmentation.jl

@@ -37,7 +37,8 @@ function reduce_augmentation(input::AbstractArray{T,N}, n) where {T<:AbstractFlo
     axs = axes(input, N)
     colons = ntuple(Returns(:), N - 1)
     for (i, ax) in enumerate(first(axs):n:last(axs))
-        view(out, colons..., i) .= sum(view(input, colons..., ax:(ax + n - 1)); dims=N) / n
+        view(out, colons..., i) .=
+            dropdims(sum(view(input, colons..., ax:(ax + n - 1)); dims=N); dims=N) / n
     end
     return out
 end