提交 588dcf8f authored 作者: Razvan Pascanu's avatar Razvan Pascanu

re-add change

上级 91f2ac84
......@@ -248,6 +248,12 @@ class Scan(PureOp):
rval = rval.astype(as_var.type.dtype)
if rval.ndim == as_var.ndim:
rval = as_var.type.filter_variable(rval)
else:
tmp = as_var.type.__class__(
broadcastable=tuple(var.broadcastable[:1])+\
tuple(as_var.broadcastable),
dtype=as_var.dtype)
rval = tmp.filter_variable(rval)
return rval
# Check if input sequences and variables representing a slice of
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论