make contiguous at end

Author: lucidrains

lambda_networks/lambda_networks.py

@@ -77,4 +77,4 @@ def forward(self, x):
 
         Y = Yc + Yp
         out = rearrange(Y, 'b (hh ww) h v -> b (h v) hh ww', hh = hh, ww = ww)
-        return out
+        return out.contiguous()

setup.py

@@ -3,7 +3,7 @@
 setup(
   name = 'lambda-networks',
   packages = find_packages(),
-  version = '0.2.0',
+  version = '0.2.1',
   license='MIT',
   description = 'Lambda Networks - Pytorch',
   author = 'Phil Wang',