提交 67713292 authored 作者: James Bergstra's avatar James Bergstra

added bounds check to 1-hot cross-entropy calculation in nnet.py

上级 bb724bc6
...@@ -581,6 +581,11 @@ class CrossentropySoftmaxArgmax1HotWithBias(gof.Op): ...@@ -581,6 +581,11 @@ class CrossentropySoftmaxArgmax1HotWithBias(gof.Op):
""", """,
inside_row_loop, inside_row_loop,
""" """
if ((y_i >= %(x)s->dimensions[1]) || (y_i < 0))
{
PyErr_SetString(PyExc_ValueError, "y_i value out of bounds");
%(fail)s;
}
nll_i[0] = - x_i[y_i*Sx] nll_i[0] = - x_i[y_i*Sx]
- b_i[y_i*Sb] - b_i[y_i*Sb]
+ row_max + row_max
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论