Yeah, the LSTM doc says you need to pass in h and c as a tuple for the hidden state 11 and gives the shapes under “Inputs”.
You only give it one Tensor and it tries to take the first dimension apart to get h and c.
Best regards
Thomas