提交 69544d41 authored 作者: Frederic's avatar Frederic

fix to error message.

上级 cb59fa4e
...@@ -3338,6 +3338,7 @@ class Rebroadcast(Op): ...@@ -3338,6 +3338,7 @@ class Rebroadcast(Op):
def c_code(self, node, nodename, inp, out, sub): def c_code(self, node, nodename, inp, out, sub):
iname, = inp iname, = inp
oname, = out oname, = out
fail = sub['fail']
if isinstance(node.inputs[0].type, TensorType): if isinstance(node.inputs[0].type, TensorType):
code = "" code = ""
for axis, value in self.axis.iteritems(): for axis, value in self.axis.iteritems():
...@@ -3346,7 +3347,7 @@ class Rebroadcast(Op): ...@@ -3346,7 +3347,7 @@ class Rebroadcast(Op):
if(PyArray_DIMS(%(iname)s)[%(axis)s] != 1){ if(PyArray_DIMS(%(iname)s)[%(axis)s] != 1){
PyErr_Format(PyExc_ValueError, PyErr_Format(PyExc_ValueError,
"Dimension %(axis)s in Rebroadcast's input was" "Dimension %(axis)s in Rebroadcast's input was"
" supposed to be 1 (got %s instead)", " supposed to be 1 (got %%d instead)",
PyArray_DIMS(%(iname)s)[%(axis)s]); PyArray_DIMS(%(iname)s)[%(axis)s]);
%(fail)s %(fail)s
} }
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论