improve example

FluxML · Oct 11, 2023 · 75744f9 · 75744f9
1 parent 78cec03
commit 75744f9
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/src/noshow.jl b/src/noshow.jl
@@ -21,11 +21,13 @@ Chain(
   Dense(9 => 10),                       # 100 parameters
 )                   # Total: 8 arrays, 145 parameters, 1.191 KiB.
 
-julia> PseudoLayer((i,o)::Pair) = Parallel(+, Dense(i => o, relu), Dense(i => o, tanh));
+julia> PseudoLayer((i,o)::Pair) = NoShow(
+                                    "PseudoLayer(\$i => \$o)",
+                                    Parallel(+, Dense(i => o, relu), Dense(i => o, tanh)),
+                                  )
+PseudoLayer (generic function with 1 method)
 
-julia> mid = PseudoLayer(3 => 10);
-
-julia> Chain(Dense(2 => 3), NoShow("PseudoLayer(3 => 10)", mid), Dense(9 => 10))
+julia> Chain(Dense(2 => 3), PseudoLayer(3 => 10), Dense(9 => 10))
 Chain(
   Dense(2 => 3),                        # 9 parameters
   PseudoLayer(3 => 10),                 # 80 parameters