gradient for prod(f, x)

#157
FluxML · Apr 15, 2019 · b3bcdf9 · b3bcdf9
1 parent 8cece0d
commit b3bcdf9
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/lib/array.jl b/src/lib/array.jl
@@ -126,6 +126,11 @@ end
   end
 end
 
+function _forward(cx::Context, ::typeof(prod), f, xs::AbstractArray)
+  y, back = forward(cx, (xs -> prod(f.(xs))), xs)
+  y, ȳ -> (nothing, nothing, back(ȳ)...)
+end
+
 @adjoint function maximum(xs; dims = :)
   max, i = findmax(xs, dims = dims)
   max, function (Δ)