TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

remove some explicit back!

Browse Source
This commit is contained in:
Mike J Innes 2017-08-18 01:18:07 +01:00
parent 052cc52ada
commit 21089fea9c
1 changed files with 0 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
src/layers/affine.jl
View File

@ -6,16 +6,3 @@ end
Affine(in::Integer, out::Integer; init = initn) = Affine(in::Integer, out::Integer; init = initn) =
Affine(init(in, out), init(1, out)) Affine(init(in, out), init(1, out))
function back!(m::Affine, Δ, x)
W, b = m.W, m.b
W.Δx[:] = x' * Δ
b.Δx[:] = sum(Δ, 1)
Δ * W.x'
end
function update!(m::Affine, η)
update!(m.W, η)
update!(m.b, η)
m
end