the inputs data need to be merged together using np.concatenate. e.g: img = np.concatenate((img, img, img, img), axis=0)