Using negative index to reshape the input tensor.

Author: rodgzilla

model_py.py

@@ -157,7 +157,7 @@ def __init__(self, cfg, vocab=40990, n_ctx=512):
         nn.init.normal_(self.embed.weight, std=0.02)
 
     def forward(self, x):
-        x = x.view(-1, x.size(2), x.size(3))
+        x = x.view(-1, x.size(-2), x.size(-1))
         e = self.embed(x)
         h = e.sum(dim=2)
         for block in self.h: