keras中的Flatten和Reshape

最新推荐文章于 2022-10-31 18:29:04 发布

转载最新推荐文章于 2022-10-31 18:29:04 发布 · 1.6k 阅读

1 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/andyniu/p/7472609.html

文章标签：

#人工智能 #python

本文探讨了在SSD模型构建中使用Flatten层的影响及其实现原理。作者通过阅读源码理解了Flatten层如何保持数据对应关系，并揭示了其底层通过Reshape函数实现的细节。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

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

转载于:https://www.cnblogs.com/andyniu/p/7472609.html