提交 bc9d27bf authored 作者: Joseph Turian's avatar Joseph Turian

Better error messages in sandbox.nnet_ups

上级 86cf354b
...@@ -273,7 +273,7 @@ class SoftmaxWithBiasDx(theano.Op): ...@@ -273,7 +273,7 @@ class SoftmaxWithBiasDx(theano.Op):
} }
if (%(dy)s->dimensions[0] != %(sm)s->dimensions[0]) if (%(dy)s->dimensions[0] != %(sm)s->dimensions[0])
{ {
PyErr_SetString(PyExc_ValueError, "dimension mismatch"); PyErr_SetString(PyExc_ValueError, "dy.shape[0] != sm.shape[0]");
%(fail)s; %(fail)s;
} }
if ((NULL == %(dx)s) if ((NULL == %(dx)s)
...@@ -548,10 +548,14 @@ class CrossentropySoftmax1HotWithBiasDx (theano.Op): ...@@ -548,10 +548,14 @@ class CrossentropySoftmax1HotWithBiasDx (theano.Op):
PyErr_SetString(PyExc_ValueError, "rank error"); PyErr_SetString(PyExc_ValueError, "rank error");
%(fail)s; %(fail)s;
} }
if ((%(dnll)s->dimensions[0] != %(sm)s->dimensions[0]) if (%(dnll)s->dimensions[0] != %(sm)s->dimensions[0])
|| (%(dnll)s->dimensions[0] != %(y_idx)s->dimensions[0]))
{ {
PyErr_SetString(PyExc_ValueError, "dimension mismatch"); PyErr_SetString(PyExc_ValueError, "dnll.shape[0] != sm.shape[0]");
%(fail)s;
}
if (%(dnll)s->dimensions[0] != %(y_idx)s->dimensions[0])
{
PyErr_SetString(PyExc_ValueError, "dnll.shape[0] != y_idx.shape[0]");
%(fail)s; %(fail)s;
} }
if ((NULL == %(dx)s) if ((NULL == %(dx)s)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论