#7

khirotaka · Aug 15, 2020 · 6dd8268 · 6dd8268
1 parent 933a574
commit 6dd8268
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/enchanter/addons/layers/positional_encoding.py b/enchanter/addons/layers/positional_encoding.py
@@ -51,11 +51,13 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
 
         Args:
             x: (N, E, L)
-            N ... batch size
-            E ... features
-            L ... seq len
+
+                - N ... batch size
+                - E ... features
+                - L ... seq len
 
         Returns:
+
             (N, E, L)
 
         """