keras image_data_format

model.add(Conv2D(32, kernel_size=(3,3), activation='relu', input_shape=(1024,1024,1), data_format='channels_first' ))

의의 식으로 있으면

channels_first = 샘플, 채널, 행, 열

channels_last = 샘플, 행, 열, 채널