Here I am using Fahion-mnist data set. Its already divided into test and train. I want each img to be a 28x28x3 shape.
X_train shape: (60000, 784) X_test sh