scalar gradients

2018-06-21 13:12:42 +01:00 · 2018-06-21 13:12:42 +01:00 · aea1e73cde
commit aea1e73cde
parent ac1448f677
1 changed files with 1 additions and 1 deletions
--- 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)