Convop infer_shape now uses the 'logical' imshp/kshp if present.

上级 9d371eee
...@@ -491,7 +491,13 @@ class ConvOp(Op): ...@@ -491,7 +491,13 @@ class ConvOp(Op):
fmo = kshp[0] fmo = kshp[0]
if self.imshp is not None and self.kshp is not None: if self.imshp is not None and self.kshp is not None:
fmshp = ConvOp.getOutputShape(self.imshp[1:], self.kshp, (self.dx,self.dy), self.out_mode) imshp = self.imshp
kshp = self.kshp
if self.imshp_logical:
imshp = self.imshp_logical
if self.kshp_logical:
kshp = self.kshp_logical
fmshp = ConvOp.getOutputShape(imshp[1:], kshp, (self.dx,self.dy), self.out_mode)
outshp = (batch_size,fmo) + tuple(fmshp) outshp = (batch_size,fmo) + tuple(fmshp)
return [outshp] return [outshp]
else: else:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论