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

问题描述: 最近需要将模型由pytorch改为keras框架,但是在conv2DTranspose这里总是出错,代码情况如下:  本来输入Conv2DTranspose的张量大小是(b,c,32,32),按照常理来计算,Conv2DTranspose的输出张量大小应该是(b,c,64,64),但是这里总是对不上,总是输出(b,c,?,?),最终进行查看也是一些莫名其妙的尺寸。 解决方案: 后面查了
相关文章
相关标签/搜索