提交 9db4b36a authored 作者: James Bergstra's avatar James Bergstra

Adjusted the output broadcastable flags from convOp to be more correct (but

still they are not perfect because there are cases where even the last two dimensions could be broadcastable).
上级 69f07b84
...@@ -192,7 +192,8 @@ class ConvOp(Op): ...@@ -192,7 +192,8 @@ class ConvOp(Op):
raise NotImplementedError("The image and the kernel must have the same type." raise NotImplementedError("The image and the kernel must have the same type."
"inputs(%s), kerns(%s)"%(_inputs.dtype, _kerns.dtype)) "inputs(%s), kerns(%s)"%(_inputs.dtype, _kerns.dtype))
output = tensor.tensor(dtype=_inputs.type.dtype, output = tensor.tensor(dtype=_inputs.type.dtype,
broadcastable=[False]*outdim); broadcastable=[_inputs.broadcastable[0],
_kerns.broadcastable[0], False, False]);
return gof.Apply(self, [_inputs, _kerns], [output]) return gof.Apply(self, [_inputs, _kerns], [output])
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论