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

编程问答 2022-08-13

问题描述

我正在关注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时，代码可以工作，这给我一个错误。

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）

google-colaboratorykeraskerasocrocrpythontensorflowtensorflowtensorflow