最近在看SSD源码的时候,就一直不理解,在模型构建的时候如果使用Flatten或者是Merge层,那么整个数据的shape就发生了变化,那么还可以对应起来么(可能你不知道我在说什么)?后来不知怎么的,就想明白了,只要先前按照同样的方式进行操作,那么就可以对应起来。同样的,只要按照之前操作的逆操作,就可以将数据的shape进行还原。
最后在说一句,在追看Tensorflow源码的时候,发现Flatten底层使用Reshape函数来实现的,和我之前预想的差不多。还有,numpy中的flatten默认是按照行来进行展开的,我是不是在其他博客里面有提到,我不记得了,算了,就这样子吧。
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:keras中的Flatten和Reshape - Python技术站