提交 e589234b authored 作者: Benjamin Scellier's avatar Benjamin Scellier

file theano/gpuarray/tests/test_abstractconv.py

上级 599eec95
...@@ -3,7 +3,7 @@ from __future__ import absolute_import, print_function, division ...@@ -3,7 +3,7 @@ from __future__ import absolute_import, print_function, division
from nose.plugins.skip import SkipTest from nose.plugins.skip import SkipTest
from nose.tools import assert_raises from nose.tools import assert_raises
import numpy import numpy as np
from theano.tensor.nnet.tests import test_abstract_conv from theano.tensor.nnet.tests import test_abstract_conv
from ..type import GpuArrayType, gpuarray_shared_constructor, get_context from ..type import GpuArrayType, gpuarray_shared_constructor, get_context
...@@ -250,5 +250,5 @@ class TestDnnConvTypes(test_abstract_conv.TestConvTypes): ...@@ -250,5 +250,5 @@ class TestDnnConvTypes(test_abstract_conv.TestConvTypes):
self.filters = gpu_ftensor4() self.filters = gpu_ftensor4()
self.topgrad = gpu_ftensor4() self.topgrad = gpu_ftensor4()
self.constant_tensor = gpuarray.array( self.constant_tensor = gpuarray.array(
numpy.zeros((3, 5, 7, 11), dtype='float32'), np.zeros((3, 5, 7, 11), dtype='float32'),
context=get_context(test_ctx_name)) context=get_context(test_ctx_name))
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论