数据不匹配报错，该如何改？

问题遇到的现象和发生背景

数据不匹配报错，该如何改？

用代码块功能插入代码，请勿粘贴截图

运行结果及报错内容

我的解答思路和尝试过的方法

我想要达到的结果

看样子是最后1个batch大小不匹配，你把数据补全成batch的倍数呗