Skip to content

Instantly share code, notes, and snippets.

@dipanjanS
Created September 20, 2019 07:47
Show Gist options
  • Select an option

  • Save dipanjanS/717292b4ebc7e6becffd3f06daf629f8 to your computer and use it in GitHub Desktop.

Select an option

Save dipanjanS/717292b4ebc7e6becffd3f06daf629f8 to your computer and use it in GitHub Desktop.
# reshape for feeding into the model
train_images_gr = train_images.reshape(train_images.shape[0], 28, 28, 1)
test_images_gr = test_images.reshape(test_images.shape[0], 28, 28, 1)
print('\nTrain_images.shape: {}, of {}'.format(train_images_gr.shape, train_images_gr.dtype))
print('Test_images.shape: {}, of {}'.format(test_images_gr.shape, test_images_gr.dtype))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment