跑了几轮batch后报错：维度不匹配

在同一个epoch里跑了几轮后，报错维度不匹配，原因是设置全连接层维度时，样本量用的batch_size = 64，但是最后取最后一个batch时，可能不够64，因此报错唯独不匹配。

解决方法：

 在定义模型时，读取输入数据的样本维度，使用该实时维度。