From f932f4bd9fd95d98cacf96743b692f8e2b3d02c1 Mon Sep 17 00:00:00 2001 From: Mike J Innes Date: Wed, 1 Feb 2017 21:10:46 +0530 Subject: [PATCH] add corrected output --- docs/src/models/debugging.md | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/docs/src/models/debugging.md b/docs/src/models/debugging.md index 5e7eed80..2db55095 100644 --- a/docs/src/models/debugging.md +++ b/docs/src/models/debugging.md @@ -49,3 +49,12 @@ This is a lot like Julia's own `code_warntype`; but instead of annotating expres This makes the problem fairly obvious. We tried to multiply the output of the first layer `(5, 20)` by a parameter `(21, 15)`; the inner dimensions should have been equal. Notice that while the first `Affine` layer is displayed as-is, the second was inlined and we see a reference to where the `W * x + b` line was defined in Flux's source code. In this way Flux makes it easy to drill down into problem areas, without showing you the full graph of thousands of nodes at once. + +With the typo fixed, the output of `@shapes` looks as follows: + +```julia +# /Users/mike/test.jl, line 18: +opossum = σ(Affine(10, 20)(Input()[1]::(5,10))::(5,20))::(5,20) +# /Users/mike/test.jl, line 19: +wren = softmax(Affine(20, 15)(opossum)::(5,15))::(5,15) +```