update readme

lucidrains · Dec 18, 2020 · c6c334a · c6c334a
1 parent 85d1bcc
commit c6c334a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -16,15 +16,15 @@ $ pip install point-transformer-pytorch
 import torch
 from point_transformer_pytorch import PointTransformerLayer
 
-x = torch.randn(1, 16, 128)
-pos = torch.randn(1, 16, 3)
-
 attn = PointTransformerLayer(
     dim = 128,
     pos_mlp_hidden_dim = 64,
     attn_mlp_hidden_mult = 4
 )
 
+x = torch.randn(1, 16, 128)
+pos = torch.randn(1, 16, 3)
+
 attn(x, pos) # (1, 16, 128)
 ```