Keras踩坑日记-1

Keras模型中，所有tensor（包含batch_size那个维度）不允许是1D的，至少是2D。如果你需要的一个变量是一个单独的数，则考虑上batch_size这一维之后会是一个1D的tensor（shape=(batch_size,)），这个时候Keras会自动地把它补成2维（shape=(batch_size, 1)），这个1是后续你需要考虑的，不然就可能出现各种奇奇怪怪的shape errors。当然，你可以在后续处理中用reshape或者squeeze把这个大小为1的维度给消灭掉。