update readme

lucidrains · lucidrains · commit ee2776f78781 · 2020-11-12T14:49:00.000-08:00
diff --git a/README.md b/README.md
@@ -28,7 +28,7 @@ hamburger = Hamburger(
 )
 
 x = torch.randn(1, 512, 32, 32)
-hamburger(x) + x # (1, 512, 1024)
+hamburger(x) + x # (1, 512, 32, 32)
 ```
 
 ## Citations
