修改后的神经网络导致损失函数输入/目标大小不匹配。预期输入 batch_size (115520) 以匹配目标 batch_size (80)

Question

我想通过删除池化层和丢弃层来修改这个神经网络架构，以便与更简单的版本进行比较。但是当我使用这个新网络运行时，批处理输入和目标不一致。这是我首先修改的代码：

待修改网络

这是我的修改版本：

错误信息

运行和可视化网络学习的功能

抱歉，我确信我遗漏了一些非常有用的、提高可读性的编码技术。提前致谢！

本质上，我只是认为删除池化层和丢弃层会以相同的方式快速而简单地工作。但我认为缺少池化层和丢失层会改变输出大小，但我不确定如何解决这个问题。