Numpy array as an input to model

Question

I have a input to the model in the format [train[:, :, 0:count]] as shown below

n_step = 10
count = 8
def create_model():
    input = Input(shape