what is the 200 hundred for. To define a hidden and cell state you should do something like this
 def init_state(self, batch_size):
        return (torch.zeros(self.num_layers, batch_size, self.hidden_size),
                torch.zeros(self.num_layers, batch_size, self.hidden_size))

then do
hs = self.init_state(batch_size)
x, hs = self.lstm(x,hs)