提交 a98609d2 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Add MakeVector case in get_vector_len, now that it may replace Join.

上级 f26fbaec
......@@ -2851,6 +2851,8 @@ def get_vector_length(v):
return join.vec_length(v)
except ValueError:
pass
if v.owner and isinstance(v.owner.op, theano.tensor.opt.MakeVector):
return len(v.owner.inputs)
if v.owner and isinstance(v.owner.op, Shape):
return v.owner.inputs[0].type.ndim
raise ValueError("length not known")
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论