update with diagram

Author: lucidrains

README.md

@@ -1,6 +1,8 @@
+<img src="./λ.png" width="500px"></img>
+
 ## Lambda Networks - Pytorch
 
-Implementation of Lambda Networks, a new approach to image recognition that reaches SOTA on ImageNet. The new method utilizes lambda layer, which captures interactions by transforming contexts into linear functions, termed lambdas, and applying these linear functions to each input separately.
+Implementation of λ Networks, a new approach to image recognition that reaches SOTA on ImageNet. The new method utilizes λ layer, which captures interactions by transforming contexts into linear functions, termed lambdas, and applying these linear functions to each input separately.
 
 ## Install
 
@@ -48,6 +50,12 @@ x = torch.randn(1, 32, 64, 64)
 layer(x) # (1, 32, 64, 64)
 ```
 
+For fun, you can also import this as follows
+
+```python
+from lambda_networks import λLayer
+```
+
 ## Todo
 
 - [x] Lambda layers with structured context

lambda_networks/__init__.py

@@ -1 +1,2 @@
 from lambda_networks.lambda_networks import LambdaLayer
+λLayer = LambdaLayer

setup.py

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