update with changes to Kernel and SphericalTensor

blondegeek · Jul 11, 2020 · b51882f · b51882f
1 parent e4c5606
commit b51882f
Showing 1 changed file with 9 additions and 39 deletions.
diff --git a/nuts_and_bolts_of_e3nn.ipynb b/nuts_and_bolts_of_e3nn.ipynb
@@ -21,7 +21,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 1,
+   "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -32,7 +32,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 2,
+   "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -92,32 +92,9 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 3,
+   "execution_count": null,
    "metadata": {},
-   "outputs": [
-    {
-     "data": {
-      "text/plain": [
-       "[<matplotlib.lines.Line2D at 0x7f14e2593750>]"
-      ]
-     },
-     "execution_count": 3,
-     "metadata": {},
-     "output_type": "execute_result"
-    },
-    {
-     "data": {
-      "image/png": "iVBORw0KGgoAAAANSUhEUgAAAXQAAAD4CAYAAAD8Zh1EAAAABHNCSVQICAgIfAhkiAAAAAlwSFlzAAALEgAACxIB0t1+/AAAADh0RVh0U29mdHdhcmUAbWF0cGxvdGxpYiB2ZXJzaW9uMy4yLjEsIGh0dHA6Ly9tYXRwbG90bGliLm9yZy+j8jraAAAgAElEQVR4nO3deXSV933n8fdXu8QitCFAAiQZAcaAsY1BGGzjHezEZI/dSbPUE7dz4kk67aTjNKmTupOmaTptz0zdNHSck8aTxHHsNCEJCdQxxICRWGxjDEIgJDBi08oiCa33N39cXVuWJXQl3Xufu3xe53DOXR7d5/vY0kePvs/v+f3MOYeIiMS+JK8LEBGR0FCgi4jECQW6iEicUKCLiMQJBbqISJxI8WrH+fn5rqSkxKvdi4jEpP379zc75wqGe8+zQC8pKWHfvn1e7V5EJCaZ2cmR3lPLRUQkTijQRUTihAJdRCROKNBFROKEAl1EJE6MGuhm9l0zazSzN0d438zsf5tZrZm9YWY3hr5MEREZTTBn6N8D1l3l/fVA+cC/R4FvT7wsEREZq1HHoTvnXjazkqtssgH4vvPPw1tpZtPMbKZz7myIahQJO+cch89e4tWTbbR19pI3OY2b5uawoHAKZuZ1eSJBCcWNRUXAqUHPGwZee0+gm9mj+M/imTNnTgh2LTJxv60+z7e21HDk3OX3vLe4aCqPr7uWNeX5HlQmMjYRvVPUObcR2AiwfPlyrawhnuru6+cr//4mP9nfQFnBJL7xoSXcPr+AginpnL/UxbYjjXzn5To+8XQVn1o1ly8/sIi0FI0jkOgVikA/Dcwe9Lx44DWRqNXe3ccfPrOPXbUtfO6Oa/jCXfPfFdbFOVn8/qoSPnbzbP72NzU8vbOet1o7+fYnbiIjNdnDykVGForTjU3AJwdGu1QAF9U/l2jW0+fjD5/ZR2VdK//ro9fzxfsWjnjmnZ6SzF+8bxHf+NASth9t4rEfvkq/T39cSnQa9QzdzH4ErAXyzawB+CqQCuCc+xdgM3A/UAt0Ap8JV7EiE+Wc4ys/O8iu2hb+7qPX8+GbioP6uodXzKG338cTPz/E139VzRPvXxTmSkXGLphRLg+P8r4DPheyikTC6IVXT/PcvgYeu2MeHwkyzAM+uaqEuqYOvrurnhWlOaxbPDNMVYqMj67wSMI42dLBV3/+JitKc/lv98wf12f8+f3XsrQ4m8d/epBzF7tCXKHIxCjQJSE45/jSTw+SlGT8w8eXkZw0vrHlaSlJ/OPHl9Hd6+MrPxv25mkRzyjQJSH86uBZXjnewp/dt4CiaZkT+qyygsl84e5yXqw+z4uHz4eoQpGJU6BL3Ovo7uN//rKa62ZN5fdWzg3JZ/7B6lLmTZ/M135xiK7e/pB8pshEKdAl7m18uY5zl7p4csPicbdahkpLSeJr77+OhrYr/KDqrZB8pshEKdAlrrV19PD0znrWL57BTXNzQvrZa8rzueWaPJ7aVkt7d19IP1tkPBToEte+83IdHT194x7VMpov3reA1o4ent5RH5bPFxkLBbrEreb2bv7tlRM8eP0s5hdOCcs+bpiTw72LCvm/O+u43NUbln2IBEuBLnHr+6+coKuvn8/fVR7W/Xzujnlc7urj2T2nRt9YJIwU6BKXrvT080zlSe5aWMg1BZPDuq/rZ09jVVkeT++sp6fPF9Z9iVyNAl3i0k9fa6Cts5fP3loakf394e1lnLvUxaYDZyKyP5HhKNAl7vh8jqd31LO0OJsVpbkR2eft8wtYOGMKT++sxz+9kUjkKdAl7mw/2khdcwePrCmN2PJxZsbvr5pL9dlLvHbqQkT2KTKUAl3izg+r3qJgSjr3L4nsbIgblhUxOT2F/1d5MqL7FQlQoEtcOXvxCi8daeSjNxWTmhzZb+/J6Sl84IZZ/PKNs7R19ER03yKgQJc485N9DfgcPHSzN4uQf6JiLj19Pp7f3+DJ/iWxKdAlbvT7HD/ee4o18/KZk5flSQ0LZ0xl+dwcfrTnLV0clYhToEvc2HGsidMXrvDwCm/OzgM+tnw2dc0dvK6LoxJhCnSJG8/vbyAnK5V7FhV6Wsf6JTPISE3ihVfVdpHIUqBLXGjv7uPF6vM8sHQmaSnefltPyUjlvutm8IsDZ+nu01zpEjkKdIkLW948R1evjw8sK/K6FAA+dGMxF6/0su1Io9elSAJRoEtc+NnrpynOyQz5nOfjtfqaPKZPSeeFV097XYokEAW6xLzGy13sqm1mw7JZEbszdDQpyUlsWDaLbUcaudCpMekSGQp0iXm/PHAWnyNq2i0B71s6iz6fY6sWkpYIUaBLzPv566dZNHMq5WFaxGK8lhZnU5yTyeaDZ70uRRKEAl1iWkNbJwcaLvLgsllel/IeZsYDS2ay81iz2i4SEQp0iWm/efMcAOuum+FxJcN7YOlMtV0kYhToEtO2HDrHwhlTKMmf5HUpw1pS5G+7/OoNtV0k/BToErMaL3ex72Qb6xdHdprcsTAzHlg6k121artI+CnQJWZtPXQe52Dd4uhstwQ8sERtF4kMBbrErC2HzlGaP4n5heFdBHqilhRlMys7g/9QoEuYBRXoZrbOzGrMrNbMHh/m/Tlmts3MXjOzN8zs/tCXKvKOC5097D7ewrrFM6LmZqKRmBl3Lypkx7Emuno1t4uEz6iBbmbJwFPAemAR8LCZLRqy2VeA55xzNwAPAf8c6kJFBnuxupE+n4va0S1D3X1tIV29PnbVNntdisSxYM7QVwC1zrk651wP8CywYcg2Dpg68DgbOBO6EkXea+uhc8zMzmBpcbbXpQSloiyPyekpvFittouETzCBXgScGvS8YeC1wb4GfMLMGoDNwH8d7oPM7FEz22dm+5qamsZRrgh09/Wzs7aZu66dHvXtloC0lCRuX1DAi9WN+HxayUjCI1QXRR8GvuecKwbuB54xs/d8tnNuo3NuuXNueUFBQYh2LYmmqq6Vzp5+7lw43etSxuSeawtputzNgQatZCThEUygnwZmD3pePPDaYI8AzwE453YDGUB+KAoUGeqlI42kpySxqiy2vsXuWDCd5CRT20XCJphA3wuUm1mpmaXhv+i5acg2bwF3AZjZtfgDXT0VCTnnHC8daWT1vHwy05K9LmdMsrNSWVGSy4uHteiFhMeoge6c6wMeA7YA1fhHsxwysyfN7MGBzf4U+KyZHQB+BHzaaclzCYPjTR281drJHTHWbgm4e1EhNecvc6q10+tSJA4F1UN3zm12zs13zl3jnPv6wGtPOOc2DTw+7Jxb7Zy73jm3zDm3NZxFS+IKLOkWa/3zgEDd24/qD1gJPd0pKjHlpSONLCicQtG0TK9LGZeSvCzm5Gbxuxq1XST0FOgSMy519bL3RGvMtlvAf9fo7fMLeOV4C919umtUQkuBLjFjx9Fm+nwuZtstAWsXFNDZ08/+E21elyJxRoEuMeOlI41kZ6Zy45xpXpcyIRVleaQlJ6mPLiGnQJeY4Jxjx7Em1pTnk5Ic29+2k9JTuLk0h9/VKNAltGL7J0MSxtHz7TRe7ua28ti6mWgka+dPp+b8Zc5cuOJ1KRJHFOgSE3Yc85/NrimPjykjbl/gP46X1XaREFKgS0zYcayZawomxexwxaHKp09mZnYGv1OgSwgp0CXqdfX2U1Xfwq1xcnYO/uGLaxcUsPNYM739Pq/LkTihQJeo9+rJNrp6fdwaJ/3zgNvnF3C5u4/XT2n2RQkNBbpEvZePNZOSZKwsy/O6lJBaVZZPkqFVjCRkFOgS9XYca+LGuTlMTk/xupSQys5KZUlRtgJdQkaBLlGtpb2bQ2cuceu8+Gq3BKyel89rb12gvbvP61IkDijQJartHDh7vXV+/FwQHWz1vHz6fI499S1elyJxQIEuUW3nsWayM/2tiXh009wc0lOS2FWrQJeJU6BL1PLf7t/M6nl5JCfFxmLQY5WRmszNJbnqo0tIKNAlah1vaufcpa64Gn8+nFvm5XHk3GWaLnd7XYrEOAW6RK2dx/xnrWvi9IJoQOD4Xjmus3SZGAW6RK3KulaKczKZnZvldSlhdd2sbLIzU9V2kQlToEtU8vkcVfUtrCyNr5uJhpOcZKwqy2NXbQtaW10mQoEuUelo42XaOnupKMv1upSIWF2ez+kLVzjZ0ul1KRLDFOgSlarqWgH/6j6JINBH36m2i0yAAl2iUmVdC0XT4r9/HlCSl8Ws7Az10WVCFOgSdfz989aEOTsH/3S6t8zLZ3ddCz6f+ugyPgp0iTrHGttp7ehJmP55wKqyPC509lJz/rLXpUiMUqBL1Kms898Gn0hn6AArB36BBY5fZKwU6BJ1Av3z4pz4WG4uWMU5WczOzVSgy7gp0CWqOOfvn68sy8UsPudvuZpVZXlU1beqjy7jokCXqPJO/zyx2i0BFQN99CPn1EeXsVOgS1QJtBtWJWigB5bZU9tFxiOoQDezdWZWY2a1Zvb4CNt8zMwOm9khM/thaMuURJGo/fOAommZzMnNUqDLuIy6SKOZJQNPAfcADcBeM9vknDs8aJty4EvAaudcm5lND1fBEr+cc1TWtbJ2fkFC9s8DKspy2XLoPD6fIylO54GX8AjmDH0FUOucq3PO9QDPAhuGbPNZ4CnnXBuAc64xtGVKIkj0/nnAqmvyuHill+pzl7wuRWJMMIFeBJwa9Lxh4LXB5gPzzWyXmVWa2brhPsjMHjWzfWa2r6mpaXwVS9yqStDx50MFZpisHJjPRiRYoboomgKUA2uBh4F/NbNpQzdyzm10zi13zi0vKIjvVWhk7CrrWpmVncHs3MTsnwfMmpbJ3Dz10WXsggn008DsQc+LB14brAHY5Jzrdc7VA0fxB7xIUPz98xYqyvISun8eUFGaxx6NR5cxCibQ9wLlZlZqZmnAQ8CmIdv8DP/ZOWaWj78FUxfCOiXO1Ta206L++dsqrslVH13GbNRAd871AY8BW4Bq4Dnn3CEze9LMHhzYbAvQYmaHgW3AF51z+ntRghZoL6xMsAm5RlJRpj66jN2owxYBnHObgc1DXnti0GMH/MnAP5Exq6xrZWZ2BnMSZP7z0czMzqQkL4vdx1t4ZE2p1+VIjNCdouI5//wt6p8PVVGWx576FvrVR5cgKdDFc8eb2mluT7z5z0dTUZbHpa4+qs+qjy7BUaCL53Yn2PqhwarQvC4yRgp08VxlXYv658OYkZ1Baf4kBboETYEunnLOUVXXwsrSxJz/fDQVZbnsqW9VH12CokAXT73TP1e7ZTgrS9VHl+Ap0MVT6p9fXWBcflW9xqPL6BTo4qmquhZmTM1gbp7658MJjEdXH12CoUAXzwTmP69I0PVDg7VS87pIkBTo4pnjTR00t3er3TKKwLwuWmdURqNAF8+8M3+LAv1q3pkfXW0XuToFunimsq6FwqnplKh/flWztM6oBEmBLp54p3+u+VuCUVGWy54T6qPL1SnQxRN1zeqfj0VFWR4XOnupOa8+uoxMgS6eqNT6oWOyUvO6SBAU6OKJyrpW9c/HoGhaJrNzM6nSghdyFQp0ibjA+qErS9U/H4uVpXlU1beojy4jUqBLxNU1d9B0Wf3zsaooy6Ots5ejjeqjy/AU6BJx7/TPtaDFWKws9f/3qjyuProMT4EuEVdZ18r0KemU5k/yupSYMjs3i6JpmZqoS0akQJeICsx/rvHn41NRlkeV5nWRESjQJaLqmztoVP983CrKcmnt6OFYY7vXpUgUUqBLRFUODLtbqf75uAR+EVbVq48u76VAl4iqrGuhYEo6Zeqfj0txTiZF0zJ1g5EMS4EuERMYf67++fiZGSvLcqmqa8U59dHl3RToEjHv9M/VbpmIitI8Wjp6qFUfXYZQoEvEBIbb6YLoxFRoXhcZgQJdImb3cfXPQ2F2biazsjPevsAsEqBAl4hQ/zx0/H10/7wu6qPLYAp0iYi6gf75KrVbQqKiLJfm9h6ON6mPLu9QoEtEaP6W0Ar00Xer7SKDBBXoZrbOzGrMrNbMHr/Kdh82M2dmy0NXosSD3cf964dq/pbQmJObxYypGVTpwqgMMmqgm1ky8BSwHlgEPGxmi4bZbgrwBaAq1EVKbNP6oaFnZlSU5VKp8egySDBn6CuAWudcnXOuB3gW2DDMdn8FfBPoCmF9EgeON7XT3K7+eahVlOXR3N7N8aYOr0uRKBFMoBcBpwY9bxh47W1mdiMw2zn3q6t9kJk9amb7zGxfU1PTmIuV2BTo82r8eWit1LwuMsSEL4qaWRLw98Cfjratc26jc265c255QUHBRHctMaLyeAszszOYq/VDQ6okL4vCqekajy5vCybQTwOzBz0vHngtYAqwGNhuZieACmCTLowKaPx5OPn76HlU1mk8uvgFE+h7gXIzKzWzNOAhYFPgTefcRedcvnOuxDlXAlQCDzrn9oWlYokpxxrbaenoUf88TFaW5tF0uZv6ZvXRJYhAd871AY8BW4Bq4Dnn3CEze9LMHgx3gRLbdh8PjD9XoIdDYFy/2i4CkBLMRs65zcDmIa89McK2aydelsSLyroWiqZlMjs30+tS4lJp/iSmT0mnsq6F31s5x+tyxGO6U1TCxudT/zzcNK+LDKZAl7A52niZts5e3e4fZhVluZy/1M2Jlk6vSxGPKdAlbNQ/jwzNjy4BCnQJm8q6FopzMpmdq/Hn4VSWP4n8yema10UU6BIePp+jqr5VwxUjQPO6SIACXcKi+twlLnT2qt0SIRVleZy71MVJ9dETmgJdwmJXbTMAq+fle1xJYghceN6ttktCU6BLWOysbWHe9MnMyM7wupSEcE3BZAqnprNz4BepJCYFuoRcd18/e+pbWKOz84gxM1bPy+eV2mZ8PvXRE5UCXULu1ZMX6Or1qd0SYWvm5dPW2cvhs5e8LkU8okCXkNtZ20RykumGoggL/EWktkviUqBLyO2sbWHZ7GlMyUj1upSEMn1qBvMLJ799QVoSjwJdQupiZy8HGy6o3eKRNfMK2FPfSldvv9eliAcU6BJSu+ta8Dl0QdQja8rz6O7zsf9km9eliAcU6BJSu2qbyUpLZtnsaV6XkpBWluaRkmTsOKa2SyJSoEtI7aptZmVpLmkp+tbywqT0FG6ck6M+eoLST52EzOkLV6hr7lD/3GNryvN588xF2jp6vC5FIkyBLiETOCtcU65A99Lqefk4B68c1zQAiUaBLiGzq7aZ/MlpLCic4nUpCe364mympKews7bJ61IkwhToEhI+n2PHsWbWzMvXcnMeS0lOouKaPN1glIAU6BISb5y+SGtHD2sXTPe6FAFuK8/nVOsV6pravS5FIkiBLiGxvaYRM7htfoHXpQhw+3z/L9btNWq7JBIFuoTE9pomlhZPI3dSmtelCDAnL4uygklsP6pATyQKdJmw1o4eDjRcYK3OzqPKHQumU1nXQmdPn9elSIQo0GXCdhxrwjlYu0CBHk3WLiigp8/Hbg1fTBgKdJmw7TVN5GSlsrRYt/tHkxWluWSmJquPnkAU6DIhPp/j5aNN3Da/gOQkDVeMJukpyayel8+2mkac0ypGiUCBLhNy8PRFWjp61G6JUmsXFNDQdoXjTR1elyIRoECXCdle0+QfrliuQI9GgV+022saPa5EIkGBLhOy/WgjS4uyyZuc7nUpMozinCzKp09WHz1BBBXoZrbOzGrMrNbMHh/m/T8xs8Nm9oaZ/dbM5oa+VIk2TZe7ef3UBe5cWOh1KXIVdyycTlV9C+3dGr4Y70YNdDNLBp4C1gOLgIfNbNGQzV4DljvnlgLPA38b6kIl+vy2+jzOwT2LFOjR7M6F0+nt91+8lvgWzBn6CqDWOVfnnOsBngU2DN7AObfNOdc58LQSKA5tmRKNth4+T9G0TK6dqdkVo9nyuTnkZKWy9dA5r0uRMAsm0IuAU4OeNwy8NpJHgF8P94aZPWpm+8xsX1OTzhZiWUd3Hztrm7n3ukLNrhjlUpKTuOvaQn57pJHefp/X5UgYhfSiqJl9AlgOfGu4951zG51zy51zywsKNCoilu041kRPn0/tlhhx76JCLnf1UVXX6nUpEkbBBPppYPag58UDr72Lmd0NfBl40DnXHZryJFptPXye7MxUVpTkel2KBOHW8gIyUpPYelhtl3gWTKDvBcrNrNTM0oCHgE2DNzCzG4Dv4A9zDXiNc339Pl460sidC6eTkqyRr7EgMy2Z28oL+I/D53XXaBwb9afROdcHPAZsAaqB55xzh8zsSTN7cGCzbwGTgZ+Y2etmtmmEj5M4sO9kGxc6e7lX7ZaYcu91Mzh7sYs3T1/yuhQJk5RgNnLObQY2D3ntiUGP7w5xXRLFth46T1pKkhaziDF3LZxOksHWw+dYUpztdTkSBvp7WcbE53NsPniW2+cXMCk9qPMBiRI5k9JYUZrLb95UHz1eKdBlTPa/1ca5S128b+lMr0uRcXhgyUyONbZTc+6y16VIGCjQZUx+eeAM6Sn+cc0Se9YvmUmSwS8OnPG6FAkDBboErd/n2PzmOe5cOJ3JarfEpPzJ6ayel88v3jij0S5xSIEuQauqb6HpcjfvWzrL61JkAt6/dBYnWzo5ePqi16VIiCnQJWi/euMsWWnJ3LlwutelyATcd90MUpNNbZc4pECXoPT2+/jNm+e469pCMtOSvS5HJiA7K5Xbygv41Rtn8fnUdoknCnQJyu9qmmjp6OHB69VuiQfvv34WZy52se9km9elSAgp0CUoz+9vIH9ymtYOjRP3XlfIpLRknt9/avSNJWYo0GVUbR09/PbIeT6wrIhUzd0SF7LSUnhg6Ux++cZZOrSSUdzQT6eMatOBM/T2Oz58k9YtiScfXT6bzp5+Nh8863UpEiIKdBnV8/sbuG7WVK6dOdXrUiSEls/NoTR/Ej/Z3+B1KRIiCnS5qsNnLnHw9EU+orPzuGNmfOSmYvbUt3KiucPrciQEFOhyVc9UniAjNYkP3aBAj0cfvrGY5CTjR3vf8roUCQEFuozo4pVefvbaGTZcX0R2VqrX5UgYzMjO4N5Fhfx47ym6evu9LkcmSIEuI3p+fwNXevv5/VVzvS5FwuhTt5RwobOXn7/+npUlJcYo0GVYPp/jmd0nuGluDouLtBhCPFtZmsvCGVP43isnNWFXjFOgy7B+d7SJEy2dfFJn53HPzPjULSVUn73E3hO6czSWKdBlWN/efpyZ2RmsX6yFLBLBB5YVMS0rlY0vH/e6FJkABbq8x94Trew50cqjt5WRlqJvkUSQmZbMZ24p5cXqRg6f0SLSsUo/rfIe/7ytltxJaTx08xyvS5EI+vQtJUxOT+Gp7bVelyLjpECXdzlw6gLbapr4zC0lmiY3wWRnpfLJVXPZfPAstY1aczQWKdDlbc45vvHravImpfGZNaVelyMeeGRNKVmpyXxrS43Xpcg4KNDlbdtrmqisa+Xzd5VrzdAElTc5nf+y9hq2HDpPVV2L1+XIGCnQBfCvSPQ3vz7C3LwsHl6h3nkie2RNGTOmZvDXm6u1olGMUaALAE/vrKfm/GW+tP5ajWxJcJlpyXzxvgUcaLjIs3u1AEYs0U+u8FZLJ//44lHuWVTIusUzvC5HosCHbixiVVke39hczbmLXV6XI0FSoCe43n4ff/zj10hJSuIvH7zO63IkSpgZf/PhJfT6fPzZC2+o9RIjFOgJ7u+21vDqWxf46w8tYda0TK/LkSgyN28Sf/G+Rbx8tIl/2qax6bFAgZ7Antt3iu/8ro6HV8zhwetneV2ORKHfWzGHD95QxD+8eFRL1cUAjU1LUJsPnuVLPz3IreX5arXIiMyMv/7gEt5q7eQLz75GRmoSdy4s9LosGUFQZ+hmts7Masys1sweH+b9dDP78cD7VWZWEupCJTT6fY5vbz/O5374KstmT+NfPnGTRrXIVWWmJfPdT9/MghlT+Oz39/O9XfWaZjdKjfqTbGbJwFPAemAR8LCZLRqy2SNAm3NuHvAPwDdDXahMTHdfPy8dOc+Hv/0K3/zNEe5fPJMf/OeVTNINRBKE7MxUfvTZCtbOL+BrvzjMxzdW8vLRJvr6fV6XJoME89O8Aqh1ztUBmNmzwAbg8KBtNgBfG3j8PPBPZmYuDL/Gn9t7io076t5+PngX79mZG/bhqF/n3vV1buT3rnJ0Qw993Pu4ytcNffdqX9fe3UdPn4+Z2Rn8/ceu54M3FGFmI5Uv8h5TMlL5108u5yf7T/GtLTV88rt7yEhNonBqBqnJSei7KXifv6uc94fhulUwgV4EDL67oAFYOdI2zrk+M7sI5AHNgzcys0eBRwHmzBnf3Yg5k9JYUDjl3S/asA8D+7zKe+P7unfv793vXv0zh36MjfyevXvL8exj8OdnpCZxc0kuaxdMV4tFxi0pyfj4zXP4wA1FvFTdyN4TbbR0dNOrM/Uxyc4Mzxq9Ef172zm3EdgIsHz58nGdvd+zqJB7FumijIiX0lOSWb9kJuuXaAGUaBLMqdppYPag58UDrw27jZmlANmAZvYREYmgYAJ9L1BuZqVmlgY8BGwass0m4FMDjz8CvBSO/rmIiIxs1JbLQE/8MWALkAx81zl3yMyeBPY55zYBTwPPmFkt0Io/9EVEJIKC6qE75zYDm4e89sSgx13AR0NbmoiIjIWGO4iIxAkFuohInFCgi4jECQW6iEicMK9GF5pZE3BynF+ez5C7UGOYjiX6xMtxgI4lWk3kWOY65wqGe8OzQJ8IM9vnnFvudR2hoGOJPvFyHKBjiVbhOha1XERE4oQCXUQkTsRqoG/0uoAQ0rFEn3g5DtCxRKuwHEtM9tBFROS9YvUMXUREhlCgi4jEiZgNdDP7KzN7w8xeN7OtZhb69ZwixMy+ZWZHBo7n381smtc1jYeZfdTMDpmZz8xicnjZaAuixwoz+66ZNZrZm17XMhFmNtvMtpnZ4YHvrS94XdN4mVmGme0xswMDx/KXId9HrPbQzWyqc+7SwOPPA4ucc3/kcVnjYmb34p9Dvs/MvgngnPsfHpc1ZmZ2LeADvgP8d+fcPo9LGpOBBdGPAvfgX2pxL/Cwc+7wVb8wCpnZbUA78H3n3GKv6xkvM5sJzHTOvWpmU4D9wAdi9P+JAZOcc+1mlgrsBL7gnGKQiNEAAAIKSURBVKsM1T5i9gw9EOYDJjHMGtGxwjm31TnXN/C0Ev+qUDHHOVftnKvxuo4JeHtBdOdcDxBYED3mOOdexr82QUxzzp11zr068PgyUI1/DeOY4/zaB56mDvwLaW7FbKADmNnXzewU8J+AJ0bbPkb8AfBrr4tIUMMtiB6T4RGPzKwEuAGo8raS8TOzZDN7HWgE/sM5F9JjiepAN7MXzezNYf5tAHDOfdk5Nxv4AfCYt9Ve3WjHMrDNl4E+/McTlYI5DpFQM7PJwAvAHw/56zymOOf6nXPL8P8VvsLMQtoOC2rFIq845+4OctMf4F9R6athLGdCRjsWM/s08D7grmhej3UM/09iUTALokuEDfSbXwB+4Jz7qdf1hIJz7oKZbQPWASG7cB3VZ+hXY2blg55uAI54VctEmdk64M+AB51znV7Xk8CCWRBdImjgQuLTQLVz7u+9rmcizKwgMILNzDLxX3wPaW7F8iiXF4AF+EdVnAT+yDkXk2dTA4trpwMtAy9VxuKIHTP7IPB/gALgAvC6c+4+b6saGzO7H/hH3lkQ/eselzQuZvYjYC3+aVrPA191zj3taVHjYGZrgB3AQfw/6wB/PrDOcUwxs6XAv+H/3koCnnPOPRnSfcRqoIuIyLvFbMtFRETeTYEuIhInFOgiInFCgS4iEicU6CIicUKBLiISJxToIiJx4v8Dc65grBTD1TkAAAAASUVORK5CYII=\n",
-      "text/plain": [
-       "<Figure size 432x288 with 1 Axes>"
-      ]
-     },
-     "metadata": {
-      "needs_background": "light"
-     },
-     "output_type": "display_data"
-    }
-   ],
+   "outputs": [],
    "source": [
     "# Inside e3nn.radial.CosineBasisModel\n",
     "import math\n",
@@ -150,7 +127,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 4,
+   "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -184,7 +161,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 5,
+   "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -201,7 +178,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 6,
+   "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -214,7 +191,7 @@
     "# we need to specify the `Rs` of the input and output\n",
     "Rs_in = [(2, 0)]\n",
     "Rs_out = [(4, 0), (4, 1), (4, 2)]\n",
-    "convolution = e3nn.point.operations.Convolution(K, Rs_in, Rs_out) "
+    "convolution = e3nn.point.operations.Convolution(K(Rs_in, Rs_out))"
    ]
   },
   {
@@ -227,7 +204,7 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 7,
+   "execution_count": null,
    "metadata": {},
    "outputs": [],
    "source": [
@@ -240,13 +217,6 @@
     "gated_act = nl.GatedBlock(\n",
     "    dimensionalities, rescaled_act.sigmoid, rescaled_act.sigmoid)"
    ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": []
   }
  ],
  "metadata": {