Replace chunk with split

Author: momohatt

time_sequence_prediction/train.py

@@ -22,7 +22,7 @@ def forward(self, input, future = 0):
         h_t2 = torch.zeros(input.size(0), 51, dtype=torch.double)
         c_t2 = torch.zeros(input.size(0), 51, dtype=torch.double)
 
-        for input_t in input.chunk(input.size(1), dim=1):
+        for input_t in input.split(1, dim=1):
             h_t, c_t = self.lstm1(input_t, (h_t, c_t))
             h_t2, c_t2 = self.lstm2(h_t, (h_t2, c_t2))
             output = self.linear(h_t2)