I am having issues using an LSTM. I get the error RuntimeError: input must have 3 dimensions, got 2. I have looked online, but I am unable to get it to work. I am fairly new to PyTorch, so any help is appreciated. Here is my code:
class TeacherUpdated(nn.Module):
    def __init__(self):
        # python requires to call ancestor's initilizer manually!
        super(TeacherUpdated, self).__init__()
        self.lstm = nn.LSTM(53, 200, 3, batch_first=True).double()
        self.linear = nn.Linear(200, 2)

    def forward(self, x):
        x, hs = self.lstm(x,200)
        x = x.reshape(-1, 200)
        x = self.linear(x)
        return x