weixin_42810302的博客

原创 PredRNN++：网络结构和代码解读

已经有很多帖子对PredRNN++的理论和改进效果进行了解读，不再赘述。直接分析结构和代码。Causal LSTM 单元三层级联结构：第一层（蓝色框）类似传统的LSTM结构用于更新时间状态C（temporal state) ；第二层（绿色框）也是类似的结构，利用更新的C和上一层的M更新空间状态M(spatial state)；第三层（灰色框）是输出门的结构，利用X、新的C和M来更新H (hidden state)。所以一个Causal LSTM 细胞单元的输入为X、H、C和M,输出为更新的H、C和M.