InvalidArgumentError：无法将张量添加到批处理中：元素数量不匹配

Question

我正在关注Keras的OCR教程here，并尝试在我的数据集上使用它
我在这部分出现错误

    train_dataset = tf.data.Dataset.from_tensor_slices((x_train, y_train))
    train_dataset = (train_dataset.map(
    encode_single_sample, num_parallel_calls=tf.data.experimental.AUTOTUNE)
    .batch(20) #batch size
    .prefetch(buffer_size=tf.data.experimental.AUTOTUNE))

我得到一个 InvalidArgumentError：无法将张量添加到批处理中：元素数量不匹配。形状为：[张量]：[55]，[批处理]：[63] [Op：IteratorGetNext] ，当我尝试遍历数据集时

    print(list(train_dataset))

当我将batch_size更改为1，但我想将其设置为20时，代码可以工作，这给我一个错误。