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

tf flatten

Browse Source
This commit is contained in:
Mike J Innes 2016-10-04 22:50:20 +01:00
parent bf04b70ad1
commit 45d30312b6
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/backend/tensorflow/tensorflow.jl
View File

@ -36,6 +36,7 @@ graph(::typeof(+), args...) = +(args...)
graph(::typeof(softmax), x) = nn.softmax(x) graph(::typeof(softmax), x) = nn.softmax(x)
graph(::typeof(relu), x) = nn.relu(x) graph(::typeof(relu), x) = nn.relu(x)
graph(::typeof(tanh), x) = tanh(x) graph(::typeof(tanh), x) = tanh(x)
graph(::typeof(flatten), x) = reshape(x, [-1])
graph(::Input, x) = x graph(::Input, x) = x