提交 cc8b517b authored 作者: Pierre Luc Carrier's avatar Pierre Luc Carrier

Convert x->nd to PyGpyArray_NDIM(x) in ops GpuSoftmax and GpuSoftmaxWithBias.

上级 116f1fa6
......@@ -476,7 +476,7 @@ class GpuSoftmax (Op):
z, = out
fail = sub['fail']
return """
if (%(x)s->nd != 2)
if (PyGpuArray_NDIM(%(x)s) != 2)
{
PyErr_SetString(PyExc_ValueError, "rank error");
%(fail)s;
......@@ -648,12 +648,12 @@ class GpuSoftmaxWithBias (GpuOp):
z, = out
fail = sub['fail']
return """
if (%(x)s->nd != 2)
if (PyGpuArray_NDIM(%(x)s) != 2)
{
PyErr_SetString(PyExc_ValueError, "rank error input");
%(fail)s;
}
if (%(b)s->nd != 1)
if (PyGpuArray_NDIM(%(b)s) != 1)
{
PyErr_SetString(PyExc_ValueError, "rank error for the bias");
%(fail)s;
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论