Merge pull request #1978 from melissatan:docs

PiperOrigin-RevId: 436510479
google · Mar 22, 2022 · e16cf72 · e16cf72
2 parents 049096b + 5f8c5f9
commit e16cf72
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/flax/linen/combinators.py b/flax/linen/combinators.py
@@ -38,9 +38,10 @@ class Foo(nn.Module):
 
       @nn.compact
       def __call__(self, x):
-        return nn.Sequential([nn.Dense(layer_size, name=f'layers_{idx}')
-                              for idx, layer_size
-                              in enumerate(self.feature_sizes)])(x)
+        return nn.Sequential([nn.Dense(4),
+                              nn.relu,
+                              nn.Dense(2),
+                              nn.log_softmax])(x)
   """
   layers: Sequence[Callable[..., Any]]