Update layers_overview.ipynb

ASEM000 · Aug 3, 2023 · fc24b38 · fc24b38
1 parent 3f5d30c
commit fc24b38
Showing 1 changed file with 23 additions and 24 deletions.
diff --git a/docs/notebooks/layers_overview.ipynb b/docs/notebooks/layers_overview.ipynb
@@ -7,17 +7,17 @@
     "# 📙 `serket.nn` layers overview\n",
     "\n",
     "\n",
-    "<table style=\"border:5px solid\">\n",
+    "<table style=\"border:1px solid\">\n",
     "\n",
     "<tr>\n",
     "\n",
-    "<td style=\"border:2px solid\" align=\"center\">\n",
+    "<td style=\"border:1px solid\" align=\"center\">\n",
     "\n",
     "[Linear](https://serket.readthedocs.io/en/latest/API/linear.html)\n",
     "\n",
     "</td>\n",
     "\n",
-    "<td style=\"border:2px solid\">\n",
+    "<td style=\"border:1px solid border-color:red\">\n",
     "\n",
     "- `Linear`, `Multilinear`, `GeneralLinear`\n",
     "- `Identity`\n",
@@ -29,13 +29,13 @@
     "\n",
     "<tr>\n",
     "\n",
-    "<td style=\"border:2px solid\" align=\"center\">\n",
+    "<td style=\"border:1px solid\" align=\"center\">\n",
     "\n",
     "[Convolution](https://serket.readthedocs.io/en/latest/API/convolution.html)\n",
     "\n",
     "</td>\n",
     "\n",
-    "<td style=\"border:2px solid\">\n",
+    "<td style=\"border:1px solid\">\n",
     "\n",
     "- `{Conv,FFTConv}{1D,2D,3D}`\n",
     "- `{Conv,FFTConv}{1D,2D,3D}Transpose`\n",
@@ -48,13 +48,13 @@
     "\n",
     "<tr>\n",
     "\n",
-    "<td style=\"border:2px solid\" align=\"center\">\n",
+    "<td style=\"border:1px solid\" align=\"center\">\n",
     "\n",
     "[Containers](https://serket.readthedocs.io/en/latest/API/containers.html)\n",
     "\n",
     "</td>\n",
     "\n",
-    "<td style=\"border:2px solid\">\n",
+    "<td style=\"border:1px solid\">\n",
     "\n",
     "- `Sequential`, `RandomApply`\n",
     "</td>\n",
@@ -63,13 +63,13 @@
     "\n",
     "<tr>\n",
     "\n",
-    "<td style=\"border:2px solid\" align=\"center\">\n",
+    "<td style=\"border:1px solid\" align=\"center\">\n",
     "\n",
     "[Pooling](https://serket.readthedocs.io/en/latest/API/pooling.html)\n",
     "\n",
     "</td>\n",
     "\n",
-    "<td style=\"border:2px solid\">\n",
+    "<td style=\"border:1px solid\">\n",
     "\n",
     "- `{Avg,Max,LP}Pool{1D,2D,3D}`\n",
     "- `Global{Avg,Max}Pool{1D,2D,3D}`\n",
@@ -81,13 +81,13 @@
     "\n",
     "<tr>\n",
     "\n",
-    "<td style=\"border:2px solid\" align=\"center\">\n",
+    "<td style=\"border:1px solid\" align=\"center\">\n",
     "\n",
     "[Reshaping](https://serket.readthedocs.io/en/latest/API/reshaping.html)\n",
     "\n",
     "</td>\n",
     "\n",
-    "<td style=\"border:2px solid\">\n",
+    "<td style=\"border:1px solid\">\n",
     "\n",
     "- `Flatten`, `Unflatten`\n",
     "- `Repeat{1D,2D,3D}`\n",
@@ -103,13 +103,13 @@
     "\n",
     "<tr>\n",
     "\n",
-    "<td style=\"border:2px solid\" align=\"center\">\n",
+    "<td style=\"border:1px solid\" align=\"center\">\n",
     "\n",
     "[Normalization](https://serket.readthedocs.io/en/latest/API/normalization.html)\n",
     "\n",
     "</td>\n",
     "\n",
-    "<td style=\"border:2px solid\">\n",
+    "<td style=\"border:1px solid\">\n",
     "\n",
     "- `{Layer,Instance,Group,Batch}Norm`\n",
     "\n",
@@ -119,13 +119,13 @@
     "\n",
     "<tr>\n",
     "\n",
-    "<td style=\"border:2px solid\" align=\"center\">\n",
+    "<td style=\"border:1px solid\" align=\"center\">\n",
     "\n",
     "[Image](https://serket.readthedocs.io/en/latest/API/image.html#)\n",
     "\n",
     "</td>\n",
     "\n",
-    "<td style=\"border:2px solid\">\n",
+    "<td style=\"border:1px solid\">\n",
     "\n",
     "- `{Avg,Gaussian}Blur2D`\n",
     "- `{Filter,FFTFilter}2D`\n",
@@ -139,13 +139,13 @@
     "\n",
     "<tr>\n",
     "\n",
-    "<td style=\"border:2px solid\" align=\"center\">\n",
+    "<td style=\"border:1px solid\" align=\"center\">\n",
     "\n",
     "[Dropout](https://serket.readthedocs.io/en/latest/API/dropout.html)\n",
     "\n",
     "</td>\n",
     "\n",
-    "<td style=\"border:2px solid\">\n",
+    "<td style=\"border:1px solid\">\n",
     "\n",
     "- `Dropout`\n",
     "- `Dropout{1D,2D,3D}`\n",
@@ -157,13 +157,13 @@
     "\n",
     "<tr>\n",
     "\n",
-    "<td style=\"border:2px solid\" align=\"center\">\n",
+    "<td style=\"border:1px solid\" align=\"center\">\n",
     "\n",
     "[Activations](https://serket.readthedocs.io/en/latest/API/activations.html)\n",
     "\n",
     "</td>\n",
     "\n",
-    "<td style=\"border:2px solid\">\n",
+    "<td style=\"border:1px solid\">\n",
     "\n",
     "- `Adaptive{LeakyReLU,ReLU,Sigmoid,Tanh}`\n",
     "- `CeLU`,`ELU`,`GELU`,`GLU`\n",
@@ -179,13 +179,13 @@
     "\n",
     "<tr>\n",
     "\n",
-    "<td style=\"border:2px solid\" align=\"center\">\n",
+    "<td style=\"border:1px solid\" align=\"center\">\n",
     "\n",
     "[Recurrent](https://serket.readthedocs.io/en/latest/API/recurrent.html)\n",
     "\n",
     "</td>\n",
     "\n",
-    "<td style=\"border:2px solid\">\n",
+    "<td style=\"border:1px solid\">\n",
     "\n",
     "- `{Dense,SimpleRNN,LSTM,GRU}Cell`\n",
     "- `{Conv,FFTConv}{LSTM,GRU}{1D,2D,3D}Cell`\n",
@@ -361,7 +361,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 7,
+   "execution_count": 3,
    "metadata": {},
    "outputs": [
     {
@@ -372,13 +372,12 @@
        "  out_features=5, \n",
        "  weight_init=he_normal, \n",
        "  bias_init=ones, \n",
-       "  dtype=<class 'jax.numpy.float16'>, \n",
        "  weight=f16[10,5](μ=0.08, σ=0.43, ∈[-1.01,0.87]), \n",
        "  bias=f16[5](μ=1.00, σ=0.00, ∈[1.00,1.00])\n",
        ")"
       ]
      },
-     "execution_count": 7,
+     "execution_count": 3,
      "metadata": {},
      "output_type": "execute_result"
     }