Colab的tpu策略时Keras预测方法的输出格式错误

Colab的tpu策略时Keras预测方法的输出格式错误