I am stuck at this weird error of dimension mismatch for yt
Perhaps your value for t in the for-loop is incorrect. Or maybe your lstm_cell_forward() is returning the wrong values for yt.