VAE LSTM shape

LSTM input and output shape: The input of the LSTM is always is a 3D array. (batch_size, time_st ...