提交 c13a44b5 authored 作者: Frederic Bastien's avatar Frederic Bastien

Better error message

上级 8e292493
...@@ -4745,6 +4745,8 @@ class Reshape(Op): ...@@ -4745,6 +4745,8 @@ class Reshape(Op):
def __init__(self, ndim, name=None): def __init__(self, ndim, name=None):
self.ndim = ndim self.ndim = ndim
if ndim < 0:
raise ValueError("The output dimensions after reshape must be 0 or greater")
assert name is None, 'name attribute for Reshape has been deprecated' assert name is None, 'name attribute for Reshape has been deprecated'
def __str__(self): def __str__(self):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论