keras中的Flatten和Reshape

Posted 2020-10-05 niudong

最近在看SSD源码的时候，就一直不理解，在模型构建的时候如果使用Flatten或者是Merge层，那么整个数据的shape就发生了变化，那么还可以对应起来么（可能你不知道我在说什么）？后来不知怎么的，就想明白了，只要先前按照同样的方式进行操作，那么就可以对应起来。同样的，只要按照之前操作的逆操作，就可以将数据的shape进行还原。

最后在说一句，在追看Tensorflow源码的时候，发现Flatten底层使用Reshape函数来实现的，和我之前预想的差不多。还有，numpy中的flatten默认是按照行来进行展开的，我是不是在其他博客里面有提到，我不记得了，算了，就这样子吧。