update readme

lucidrains · lucidrains · commit 06a48f2a5b41 · 2020-10-19T19:00:03.000-07:00
diff --git a/README.md b/README.md
@@ -74,8 +74,8 @@ layer = LambdaLayer(
     dim_u = 1
 )
 
-x = tf.random.normal((1, 64, 64, 16))
-layer(x)
+x = tf.random.normal((1, 64, 64, 16)) # channel last format
+layer(x) # (1, 64, 64, 32)
 ```
 
 ## Citations
diff --git a/lambda_networks/tfkeras.py b/lambda_networks/tfkeras.py
@@ -8,7 +8,6 @@
 def exists(val):
     return val is not None
 
-
 def default(val, d):
     return val if exists(val) else d
 

Original file line number	Diff line number	Diff line change
`@@ -74,8 +74,8 @@ layer = LambdaLayer(`
`74`	`74`	`dim_u = 1`
`75`	`75`	`)`
`76`	`76`
`77`		`-x = tf.random.normal((1, 64, 64, 16))`
`78`		`-layer(x)`
	`77`	`+x = tf.random.normal((1, 64, 64, 16)) # channel last format`
	`78`	`+layer(x) # (1, 64, 64, 32)`
`79`	`79`	```
`80`	`80`
`81`	`81`	`## Citations`