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

Merge pull request #63 from JobJob/rnncell-args

Browse Source 
Enables passing an activation function to RNN/RNNCell
This commit is contained in:
Mike J Innes 2017-09-12 13:10:43 +01:00 committed by GitHub
commit 6728295355
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/layers/recurrent.jl
View File

@ -34,8 +34,8 @@ struct RNNCell{D,V}
h::V
end
RNNCell(in::Integer, out::Integer; init = initn) =
RNNCell(Dense(in+out, out, init = initn), param(initn(out)))
RNNCell(in::Integer, out::Integer, σ=identity; init = initn) =
RNNCell(Dense(in+out, out, σ, init = init), param(init(out)))
function (m::RNNCell)(h, x)
h = m.d(combine(x, h))