Correct if I'm wrong but in LSTM(dim_number), here dim_number is for num_units in hidden cell. You have taken it as num of time steps.

Thanks