broadcast adjoint for unary minus (#1287)

FluxML · Aug 15, 2022 · 17a5673 · 17a5673
1 parent e4fc02c
commit 17a5673
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/lib/broadcast.jl b/src/lib/broadcast.jl
@@ -80,6 +80,8 @@ unbroadcast(x::AbstractArray, x̄::Nothing) = nothing
 
 @adjoint broadcasted(::typeof(-), x::Numeric, y::Numeric) = x .- y,
   Δ -> (nothing, unbroadcast(x, Δ), _minus(unbroadcast(y, Δ)))
+@adjoint broadcasted(::typeof(-), x::Numeric) = .-x,
+  Δ -> (nothing, _minus(Δ))
 _minus(Δ) = -Δ
 _minus(::Nothing) = nothing