From aea1e73cdede14a579670a59859d7f2654824e8b Mon Sep 17 00:00:00 2001 From: Mike J Innes Date: Thu, 21 Jun 2018 13:12:42 +0100 Subject: [PATCH] scalar gradients --- src/tracker/numeric.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/tracker/numeric.jl b/src/tracker/numeric.jl index 272f9ba4..755e1f7d 100644 --- a/src/tracker/numeric.jl +++ b/src/tracker/numeric.jl @@ -1,4 +1,4 @@ -function gradient(f, xs::AbstractArray...) +function gradient(f, xs...) xs = param.(xs) back!(f(xs...)) grad.(xs)