batch_shape causes multi gpu to fail

if you change 

```
batch_shape=(256,256,3)
inputs = Input(batch_shape=(5, 256, 256, 3))
```

to

```
shape=(256,256,3)
inputs = Input(shape)
```

The multi gpu setups can run.  Also, then
the generator is the only place where the batch size is specified.

