keras中的Flatten和Reshape

本文探讨了在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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值