提交 9de86161 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Stop painting over broadcastable differences (since this could lead to a bug).

上级 5f2c9126
...@@ -400,12 +400,7 @@ def expand(tensor_var, size): ...@@ -400,12 +400,7 @@ def expand(tensor_var, size):
shapes = [tensor_var.shape[x] for x in xrange(tensor_var.ndim)] shapes = [tensor_var.shape[x] for x in xrange(tensor_var.ndim)]
zeros_shape = [size + shapes[0]] + shapes[1:] zeros_shape = [size + shapes[0]] + shapes[1:]
empty = tensor.zeros(zeros_shape, empty = tensor.zeros(zeros_shape,
dtype=tensor_var.dtype) dtype=tensor_var.dtype)
# Make sure to reuse the broadcast pattern of the original tensor for
# every dimension but the first one.
broadcastable = (False,) + tensor_var.broadcastable[1:]
empty = tensor.patternbroadcast(empty, broadcastable)
return tensor.set_subtensor(empty[:shapes[0]], tensor_var) return tensor.set_subtensor(empty[:shapes[0]], tensor_var)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论