提交 ed32cd9c authored 作者: Frederic Bastien's avatar Frederic Bastien

Fix broadcast pattern problem introduced in 58c3d08e

上级 d66cefe7
......@@ -695,10 +695,10 @@ class CorrMM_gradWeights(BaseCorrMM):
height_width = [as_tensor_variable(shape[0]).astype('int64'), as_tensor_variable(shape[1]).astype('int64')]
if self.unshared is True:
broadcastable = [topgrad.type.broadcastable[0], False, False,
broadcastable = [topgrad.type.broadcastable[1], False, False,
img.type.broadcastable[1], False, False]
else:
broadcastable = [topgrad.type.broadcastable[0], img.type.broadcastable[1],
broadcastable = [topgrad.type.broadcastable[1], img.type.broadcastable[1],
False, False]
dtype = img.type.dtype
return Apply(self, [img, topgrad] + height_width,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论