Could you please explain to me the input shape and output shape here in the policy code?
Where did you find this LSTM model policy?