如何为tpu提供多个输入的Tensorflow数据？

Question

我试图为Keras（Tenserflow）提供多个输入（2个输入）的数据，以进行TPU培训，但我得到以下错误：

ValueError: The dataset returned a non-Tensor type
  ((<class 'tensorflow.python.framework.ops.Tensor'>,
    <class 'tensorflow.python.framework.ops.Tensor'>)) at index 0

我尝试了这个链接：tf.data with multiple inputs / outputs in Keras

^{pr2}$ ^{3}$