diff --git a/base/reducedim.jl b/base/reducedim.jl index fe3adef035dec..a932e826bd10c 100644 --- a/base/reducedim.jl +++ b/base/reducedim.jl @@ -651,6 +651,8 @@ for (fname, _fname, op) in [(:sum, :_sum, :add_sum), (:prod, :_prod, # User-facing methods with keyword arguments @inline ($fname)(a::AbstractArray; dims=:) = ($_fname)(a, dims) @inline ($fname)(f, a::AbstractArray; dims=:) = ($_fname)(f, a, dims) + @inline ($fname)(f, a::AbstractArray, b::AbstractArray; dims=:) = + ($_fname)(((a,b),)->f(a,b), zip(a,b), dims) # Underlying implementations using dispatch ($_fname)(a, ::Colon) = ($_fname)(identity, a, :)