提交 4c30a32b authored 作者: James Bergstra's avatar James Bergstra

more helpful error to catch symbolic shape vecs in infer_shape

上级 1ddd6f3a
......@@ -788,6 +788,8 @@ class ShapeFeature(object):
if s is None:
self.shape_of[r] = s
else:
if not isinstance(s, (tuple, list)):
raise TypeError('shapes must be tuple/list', (r, s))
if r.ndim != len(s):
raise AssertionError(
"Something inferred a shape with %d dimensions "
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论