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的维度给消灭掉。
Keras踩坑日记-1
最新推荐文章于 2019-11-22 18:31:21 发布