minor update to Readme - fixed import

Deepak-Rai-1027 · Dec 1, 2018 · 5fae39b · 5fae39b
1 parent 59b21f0
commit 5fae39b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/README.md b/README.md
@@ -91,7 +91,7 @@ Example of using layers within a model:
 ```python
 # example given for pytorch, but code in other frameworks is almost identical  
 from torch.nn import Sequential, Conv2d, MaxPool2d, Linear, ReLU
-from einops.layers.torch import Reduce
+from einops.layers.torch import Rearrange
 
 model = Sequential(
     Conv2d(3, 6, kernel_size=5),