add einops

Author: nathanhubens

fasterai/sparse/sparsifier.py

@@ -12,6 +12,7 @@
 from fastcore.basics import store_attr, listify, true
 from typing import Callable, Optional, Union, List, Tuple, Type
 from ..core.criteria import *
+from einops import rearrange
 
 # %% ../../nbs/sparse/sparsifier.ipynb 5
 class Sparsifier():

nbs/sparse/sparsifier.ipynb

@@ -46,7 +46,8 @@
     "from itertools import cycle\n",
     "from fastcore.basics import store_attr, listify, true\n",
     "from typing import Callable, Optional, Union, List, Tuple, Type\n",
-    "from fasterai.core.criteria import *"
+    "from fasterai.core.criteria import *\n",
+    "from einops import rearrange"
    ]
   },
   {