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

tweak note

Browse Source
This commit is contained in:
Mike J Innes 2016-08-30 12:52:09 +01:00
parent 0ebccb2957
commit d58fefb972
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/shape.jl
View File

@ -1,7 +1,5 @@
export Input export Input
# Shim for kicking off shape inference
typealias Dims{N} NTuple{N,Int} typealias Dims{N} NTuple{N,Int}
dims(d::Dims) = d dims(d::Dims) = d
@ -11,6 +9,8 @@ dims(i...) = (i...,)
single(i) = i single(i) = i
single(i::Dims) = length(i) == 1 ? first(i) : i single(i::Dims) = length(i) == 1 ? first(i) : i
# Shim for kicking off shape inference
type Input{N} <: Model type Input{N} <: Model
dims::Dims{N} dims::Dims{N}
end end