提交 ae781710 authored 作者: Frederic's avatar Frederic

small fix from code review

上级 0ad8e57b
...@@ -352,7 +352,7 @@ class GpuCrossentropySoftmax1HotWithBiasDx(Op): ...@@ -352,7 +352,7 @@ class GpuCrossentropySoftmax1HotWithBiasDx(Op):
%(fail)s; %(fail)s;
} }
if (PyGpuArray_DIMS(%(sm)s)[0] != if (PyGpuArray_DIMS(%(sm)s)[0] !=
PyGpuArray__DIMS(%(y_idx)s)[0]) PyGpuArray_DIMS(%(y_idx)s)[0])
{ {
PyErr_SetString(PyExc_ValueError, PyErr_SetString(PyExc_ValueError,
"sm.shape[0] != y_idx.shape[0]"); "sm.shape[0] != y_idx.shape[0]");
......
...@@ -7,7 +7,7 @@ from theano.tests.unittest_tools import SkipTest ...@@ -7,7 +7,7 @@ from theano.tests.unittest_tools import SkipTest
from theano.tensor.tests import test_basic from theano.tensor.tests import test_basic
import theano.sandbox.gpuarray import theano.sandbox.gpuarray
from theano.sandbox.gpuarray import basic_ops from .. import basic_ops
from ..type import GpuArrayType, gpuarray_shared_constructor from ..type import GpuArrayType, gpuarray_shared_constructor
from ..basic_ops import (GpuAlloc, GpuReshape, gpu_alloc, from ..basic_ops import (GpuAlloc, GpuReshape, gpu_alloc,
gpu_from_host, host_from_gpu) gpu_from_host, host_from_gpu)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论