fixed `input_shape` error
Merge pull request #1287 from jych/master_jych

fixed `input_shape` error