official/mnist: Add input_shape parameter to initial layer (#4104) (#4118)
official/mnist: Add input_shape parameter to initial layer (#4104) (#4118)


tombstone/models,cshallue/models,tombstone/models,derekjchow/models,cshallue/models,derekjchow/models,tombstone/models,alexgorban/models,derekjchow/models,derekjchow/models,tombstone/models,alexgorban/models,cshallue/models,tombstone/models,alexgorban/models,alexgorban/models,alexgorban/models,cshallue/models,derekjchow/models,tombstone/models,cshallue/models