提交 5bcefac4 authored 作者: James Bergstra's avatar James Bergstra

ConvOp - removed references to default_infer_shape

上级 4d82d018
...@@ -502,14 +502,17 @@ class ConvOp(Op): ...@@ -502,14 +502,17 @@ class ConvOp(Op):
imshp = self.imshp_logical imshp = self.imshp_logical
if self.kshp_logical: if self.kshp_logical:
kshp = self.kshp_logical kshp = self.kshp_logical
fmshp = ConvOp.getOutputShape(imshp[1:], kshp, (self.dx,self.dy), self.out_mode) try:
fmshp = ConvOp.getOutputShape(imshp[1:], kshp, (self.dx,self.dy), self.out_mode)
except TypeError:
raise NotImplementedError()
outshp = (batch_size,fmo) + tuple(fmshp) outshp = (batch_size,fmo) + tuple(fmshp)
return [outshp] return [outshp]
else: else:
# Haven't implemented this case. imshp and kshp may be symbollic # Haven't implemented this case. imshp and kshp may be symbollic
# and ConvOp.getOutputShape doesn't handle this. In this case # and ConvOp.getOutputShape doesn't handle this. In this case
# we simply let the default function do its work. # we simply let the default function do its work.
return node.env.shape_feature.default_infer_shape(node, ishapes) raise NotImplementedError()
def perform(self,node, (img2d, filtersflipped), (z,)): def perform(self,node, (img2d, filtersflipped), (z,)):
""" """
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论