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

more correct

Browse Source
This commit is contained in:
Mike J Innes 2017-01-27 17:03:22 +05:30
parent a71c79e920
commit 3b3a088851
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -20,10 +20,10 @@ 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(node(args)...) @eval graph(::typeof($op), args...) = $op(args...)
end end
graph(::typeof(.-), args...) = -(node(args)...) graph(::typeof(.-), args...) = -(args...)
# reshape hack due to https://github.com/malmaud/TensorFlow.jl/issues/79 # reshape hack due to https://github.com/malmaud/TensorFlow.jl/issues/79
batchsize(x::Tensor) = reduce_sum(slice(TensorFlow.shape(x), [0], [1])) batchsize(x::Tensor) = reduce_sum(slice(TensorFlow.shape(x), [0], [1]))
@ -51,7 +51,7 @@ interp{T<:AArray}(ctx, p::Constant{Flux.Param{T}}) =
ctx[:params][p.value] : ctx[:params][p.value] :
(ctx[:params][p.value] = Variable(convertel(Float32, p.value.x))) (ctx[:params][p.value] = Variable(convertel(Float32, p.value.x)))
interp(ctx, p::Constant) = p.value interp(ctx, p::Constant) = node(p.value)
function interp(ctx, model, args...) function interp(ctx, model, args...)
g = Flux.graph(model) g = Flux.graph(model)