Hey @wasiahmad, @smth , Can you share the code snippet on how you got all the input states?
I’m providing my code. Can you suggest my how should I go on?
Although, it is in torch, I guess the idea should be similar.

inputSize=5
hiddenSize=5
rho=5
lstm=nn.LSTM(inputSize, hiddenSize,rho)
input=torch.randn(10,5)–I am providing 10*5 input tensor
print(lstm:getHiddenState(1)) --prints the cell state and hidden state for the first time step