keras下conv2DTranspose输出张量尺寸bug的修复

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/xiaoxifei/article/details/82698725

问题描述：

最近需要将模型由pytorch改为keras框架，但是在conv2DTranspose这里总是出错，代码情况如下：

 本来输入Conv2DTranspose的张量大小是(b,c,32,32),按照常理来计算，Conv2DTranspose的输出张量大小应该是(b,c,64,64),但是这里总是对不上，总是输出(b,c,?,?),最终进行查看也是一些莫名其妙的尺寸。

解决方案：

后面查了下，认为是keras的版本所导致的，出于稳定考虑，将所有层从tensorflow.python下的keras进行调入即可解决这个问题，

 像这样处理，而不是使用 import keras.layers import Conv2DTranspose

即可获得解决。