I lost 2-3 days of debugging due to this formulation
So, I tried to use something like
X = rnn_forward(single_example_ix, [None] + single_example_ix, a_prev, parameters)
It was really confusing for me when I recall, that I used rnn_forward() in optimize(), that was used in graded model().