training datasets of variable time steps

[jeshraghian]

• currently, the initialization of state variables step is only equipped for one common time step

possible solutions:

• modifying the init step to adjust to the input first dimension
• zero-extension along the time axis (sloppy fix, and might need a warning displayed to inform the user). Might be best for user to implement this themselves. Thoughts are welcome.