i am new to all deep learning, so i am sorry if my question does not make sense, i have a dataset and i would like to train it on a module, how to know the data structure for Example, mnist dataset. input data structure normally will be : [batch_size, channel, image_width, image_height] output : [batch_size, 10] since we have 10 classes. how to know this info for another dataset  ?