提交 2c1eb94b authored 作者: Frederic's avatar Frederic

all code simplification.

上级 a7f9909e
......@@ -690,8 +690,8 @@ class ConvOp(OpenMPOp):
fulloutshp = tuple(ConvOp.getOutputShape(imshp_logical[
1:], kshp_logical, (1, 1), self.out_mode))
if z[0] is None or z[0].shape != (bsize,) + (nkern,) + fulloutshp:
z[0] = numpy.zeros((bsize,) + (nkern,) + fulloutshp,
if z[0] is None or z[0].shape != (bsize, nkern,) + fulloutshp:
z[0] = numpy.zeros((bsize, nkern,) + fulloutshp,
dtype=img2d.dtype)
zz = z[0]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论