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

fix the build

Browse Source
This commit is contained in:
Mike J Innes 2017-01-16 01:21:45 +01:00
parent 271d06a376
commit d7fe525f4d
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/compiler/shape.jl
View File

@ -46,5 +46,3 @@ end
# TODO: make correct # TODO: make correct
infer(::typeof(+), a, b) = a infer(::typeof(+), a, b) = a
infer(::typeof(softmax), x) = x
infer(::typeof(σ), x) = x

3
src/layers/activation.jl
View File

@ -13,3 +13,6 @@ softmax(xs) = exp.(xs) ./ sum(exp.(xs))
flatten(xs) = reshape(xs, length(xs)) flatten(xs) = reshape(xs, length(xs))
shape(::typeof(flatten), in) = prod(in) shape(::typeof(flatten), in) = prod(in)
infer(::typeof(softmax), x) = x
infer(::typeof(σ), x) = x