Input shape attribute in models (#292)

Setting this attribute will become a requirement for some upcoming features

* Add utility function to set input_shape attribute in models
* Set this attribute in our classification models factory
  (models.create_model())
* Add shape parameter in get_dummy_input() (in addition to dataset)