Skip to content

Commit 666d6b5

Browse files
author
Michael Abbott
committed
ok, ok.
1 parent d9b7b51 commit 666d6b5

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/outputsize.jl

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ which should work out of the box for custom layers.
6262
If `m` is a `Tuple` or `Vector`, its elements are applied in sequence, like `Chain(m...)`.
6363
6464
# Examples
65-
```jldoctest
65+
```julia-repl
6666
julia> using Flux: outputsize
6767
6868
julia> outputsize(Dense(10, 4), (10,); padbatch=true)
@@ -81,7 +81,8 @@ julia> outputsize(m, (10, 10, 3, 64))
8181
8282
julia> try outputsize(m, (10, 10, 7, 64)) catch e println(e) end
8383
┌ Error: layer Conv((3, 3), 3=>16), index 1 in Chain, gave an error with input of size (10, 10, 7, 64)
84-
[...]
84+
└ @ Flux ~/.julia/dev/Flux/src/outputsize.jl:114
85+
DimensionMismatch("Input channels must match! (7 vs. 3)")
8586
8687
julia> outputsize([Dense(10, 4), Dense(4, 2)], (10, 1)) # Vector of layers becomes a Chain
8788
(2, 1)

0 commit comments

Comments
 (0)