提交 f02bb033 authored 作者: Frederic's avatar Frederic

Better error message in get_vector_length()

上级 60e0ed1c
...@@ -4230,7 +4230,8 @@ def get_vector_length(v): ...@@ -4230,7 +4230,8 @@ def get_vector_length(v):
""" """
v = as_tensor_variable(v) v = as_tensor_variable(v)
if v.ndim != 1: if v.ndim != 1:
raise TypeError('argument must be symbolic vector') raise TypeError("argument must be symbolic vector, got '%s'" %
v)
if v.type.broadcastable[0]: if v.type.broadcastable[0]:
return 1 return 1
if isinstance(v, gof.Constant) and v.type.ndim == 1: if isinstance(v, gof.Constant) and v.type.ndim == 1:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论