Skip to content

Commit

Permalink
Fix item lists for rendering by Sphinx notebook extension.
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 322212008
  • Loading branch information
j2i2 authored and Copybara-Service committed Jul 20, 2020
1 parent 21fba15 commit 367e2ea
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions trax/layers/intro.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -342,6 +342,7 @@
"\n",
"Many layer types have creation-time parameters for flexibility. The \n",
"`Concatenate` layer type, for instance, has two optional parameters:\n",
"\n",
"* `axis`: index of axis along which to concatenate the tensors; default value of -1 means to use the last axis.\n",
"* `n_items`: number of tensors to join into one by concatenation; default value is 2.\n",
"\n",
Expand Down Expand Up @@ -922,6 +923,7 @@
"\n",
"The Trax runtime supports the concept of multiple data streams, which gives\n",
"individual layers flexibility to:\n",
"\n",
" - process a single data stream ($n_{in} = n_{out} = 1$),\n",
" - process multiple parallel data streams ($n_{in} = n_{out} = 2, 3, ... $),\n",
" - split or inject data streams ($n_{in} \u003c n_{out}$), or\n",
Expand All @@ -936,6 +938,7 @@
" )\n",
"```\n",
"In other words, layer by layer:\n",
"\n",
" - `Branch(shortcut, layers)`: makes two copies of the single incoming data stream, passes one copy via the shortcut (typically a no-op), and processes the other copy via the given layers, applied in series. [$n_{in} = 1$, $n_{out} = 2$]\n",
" - `Add()`: combines the two streams back into one by adding two tensors elementwise. [$n_{in} = 2$, $n_{out} = 1$]"
]
Expand Down

0 comments on commit 367e2ea

Please sign in to comment.