提交 d3358072 authored 作者: Vikram's avatar Vikram

3D Broadcastable error fixed

上级 d051a930
...@@ -1787,7 +1787,7 @@ class GpuCorr3dMM_gradInputs(BaseGpuCorr3dMM): ...@@ -1787,7 +1787,7 @@ class GpuCorr3dMM_gradInputs(BaseGpuCorr3dMM):
broadcastable = [topgrad.type.broadcastable[0], False, broadcastable = [topgrad.type.broadcastable[0], False,
False, False, False] False, False, False]
else: else:
broadcastable = [topgrad.type.broadcastable[0], kern.type.broadcastable[-3], broadcastable = [topgrad.type.broadcastable[0], kern.type.broadcastable[-4],
False, False, False] False, False, False]
return Apply(self, [kern, topgrad] + height_width_depth, return Apply(self, [kern, topgrad] + height_width_depth,
[GpuArrayType(dtype=topgrad.dtype, [GpuArrayType(dtype=topgrad.dtype,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论