Skip to content

Commit 54183d1

Browse files
committed
separate methods for Params
1 parent 4b7524e commit 54183d1

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

src/compiler/interface.jl

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -125,7 +125,15 @@ julia> haskey(g, z) # only x and y are parameters
125125
false
126126
```
127127
"""
128-
gradient
128+
function gradient(f, ps::Params)
129+
y, back = pullback(f, ps)
130+
back(sensitivity(y))
131+
end
132+
133+
function withgradient(f, ps::Params)
134+
y, back = pullback(f, ps)
135+
(val = y, grad = back(sensitivity(y)))
136+
end
129137

130138
"""
131139
Params([A, B])

0 commit comments

Comments
 (0)