How would you configure your data if you had a inputs data set with the following dimension (batch_size=256, channels=2, height=64, width=64)  to match the LSTM input format (sequence_length, batch_size, input_size)  ?