initial commit

FluxML · Jan 14, 2024 · 4229760 · 4229760
1 parent 392a1f9
commit 4229760
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/lib/lib.jl b/src/lib/lib.jl
@@ -22,7 +22,7 @@ accum(x, y) =
 accum(x, y, zs...) = accum(accum(x, y), zs...)
 
 accum(x::Tuple, ys::Tuple...) = map(accum, x, ys...)
-accum(x::AbstractArray, ys::AbstractArray...) = accum.(x, ys...)
+accum(x::AbstractArray, ys::AbstractArray...) = Base.broadcast_preserving_zero_d(accum, x, ys...)
 
 @generated function accum(x::NamedTuple, y::NamedTuple)
   # assumes that y has no keys apart from those also in x