1. 程式人生 > >keras中的Flatten和Reshape

keras中的Flatten和Reshape

其他 個數 如果 tensor res sha 模型 flatten 函數

最近在看SSD源碼的時候,就一直不理解,在模型構建的時候如果使用Flatten或者是Merge層,那麽整個數據的shape就發生了變化,那麽還可以對應起來麽(可能你不知道我在說什麽)?後來不知怎麽的,就想明白了,只要先前按照同樣的方式進行操作,那麽就可以對應起來。同樣的,只要按照之前操作的逆操作,就可以將數據的shape進行還原。

最後在說一句,在追看Tensorflow源碼的時候,發現Flatten底層使用Reshape函數來實現的,和我之前預想的差不多。還有,numpy中的flatten默認是按照行來進行展開的,我是不是在其他博客裏面有提到,我不記得了,算了,就這樣子吧。

keras中的Flatten和Reshape