it depends on how you actually order your data,if its on a channel first basis then you should reshape your data: x_train=x_train.reshape(x_train.shape[0],channel,width,height)
if its channel last: x_train=s_train.reshape(x_train.shape[0],width,height,channel)