Merge pull request #73 from JiaqiLiu/master

Fix: Doc for LockedDropout Input Shape

torchnlp/nn/lock_dropout.py

@@ -52,7 +52,7 @@ def __init__(self, p=0.5):
     def forward(self, x):
         """
         Args:
-            x (:class:`torch.FloatTensor` [batch size, sequence length, rnn hidden size]): Input to
+            x (:class:`torch.FloatTensor` [sequence length, batch size, rnn hidden size]): Input to
                 apply dropout too.
         """
         if not self.training or not self.p: