提交 0eeba314 authored 作者: Frederic Bastien's avatar Frederic Bastien

Fix broadcast in the gpu code too.

上级 ed32cd9c
...@@ -1033,7 +1033,7 @@ class GpuCorrMM_gradWeights(BaseGpuCorrMM): ...@@ -1033,7 +1033,7 @@ class GpuCorrMM_gradWeights(BaseGpuCorrMM):
assert shape[1].ndim == 0 assert shape[1].ndim == 0
if self.unshared: if self.unshared:
broadcastable = [topgrad.type.broadcastable[0], False, False, broadcastable = [topgrad.type.broadcastable[1], False, False,
img.type.broadcastable[1], False, False] img.type.broadcastable[1], False, False]
else: else:
broadcastable = [topgrad.type.broadcastable[1], img.type.broadcastable[1], broadcastable = [topgrad.type.broadcastable[1], img.type.broadcastable[1],
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论