A Keras model (link here, for the sake of MWE) needs to predict a lot of test data, in parallel.
I define a cube as a 3D numpy.ndarray
numpy.ndarray