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

fixes

Browse Source
This commit is contained in:
Mike J Innes 2017-03-27 18:23:22 +01:00
parent 1a5e050a88
commit 05ac3855f4
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/backend/tensorflow/graph.jl
View File

@ -1,6 +1,7 @@
using Base: @get! using Base: @get!
using DataFlow: Constant, constant, Split using DataFlow: Constant, constant, Split
using DataFlow.Interpreter using DataFlow.Interpreter
using DataFlow.Interpreter: stack
using Flux: imap using Flux: imap
using TensorFlow: RawTensor, TFException using TensorFlow: RawTensor, TFException
@ -19,7 +20,7 @@ graph(::typeof(σ), x) = nn.sigmoid(x)
graph(::typeof(hcat), xs...) = concat(1, xs) graph(::typeof(hcat), xs...) = concat(1, xs)
graph(::typeof(seq), xs, n) = TensorFlow.unpack(xs, num = n, axis = 1) graph(::typeof(seq), xs, n) = TensorFlow.unpack(xs, num = n, axis = 1)
for op in (tanh, *, .*, .+, .-) for op in (tanh, *, .*, .+)
@eval graph(::typeof($op), args...) = $op(args...) @eval graph(::typeof($op), args...) = $op(args...)
end end