The x is the data being passed into the forward function. self.lstm = nn.LSTM(53, 200, 3, batch_first=True).double() is input size, hidden layers, and num of layers respectively. There is something wrong with x, hs = self.lstm(x,200) and I may not be calling this correctly.