Function to_multi_gpu

Given a keras [model], return an equivalent model which parallelizes the computation over [n_gpus] GPUs. Each GPU gets a slice of the input batch, applies the model on that slice and later the outputs of the models are concatenated to a single tensor, hence the user sees a model th

(model, n_gpus=2)