clean up notebook, add colab link.

Author: parrt

testing/examples.ipynb

@@ -6,6 +6,8 @@
    "source": [
     "# TensorSensor\n",
     "\n",
+    "[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/parrt/tensor-sensor/blob/sbrugman-patch-2/testing/examples.ipynb)\n",
+    "\n",
     "TensorSensor clarifies exceptions by augmenting messages and visualizing Python code to indicate the shape of tensor variables (see figure to the right for a teaser). It works with Tensorflow, PyTorch, JAX, and Numpy, as well as higher-level libraries like Keras and fastai.  Besides trapping exceptions, TensorSensor can provide visualizations of code that executes without exception. You can also see the complete expression tree computation and the shapes of self-expression partial results.\n",
     "\n",
     "See article [Clarifying exceptions and visualizing tensor operations in deep learning code](https://explained.ai/tensor-sensor/index.html) at [explained.ai](https://explained.ai)."
@@ -1540,7 +1542,7 @@
    "outputs": [],
    "source": [
     "import tsensor\n",
-    "with tsensor.explain(legend=True) as explained:\n",
+    "with tsensor.explain() as explained:\n",
     "    a = torch.relu(x)\n",
     "    b = W @ b + torch.zeros(2000,1)+(h+3).dot(h)"
    ]