提交 5a2675fd authored 作者: Benjamin Scellier's avatar Benjamin Scellier

file theano/gpuarray/tests/test_blas.py

上级 57c8ffdc
...@@ -2,7 +2,7 @@ from __future__ import absolute_import, print_function, division ...@@ -2,7 +2,7 @@ from __future__ import absolute_import, print_function, division
from unittest import TestCase from unittest import TestCase
from nose.plugins.skip import SkipTest from nose.plugins.skip import SkipTest
import itertools import itertools
import numpy import numpy as np
import theano import theano
from theano import tensor from theano import tensor
...@@ -138,8 +138,8 @@ def test_gemv_zeros(): ...@@ -138,8 +138,8 @@ def test_gemv_zeros():
# Apply to an empty matrix shape (5,0) and an empty vector shape (0,) # Apply to an empty matrix shape (5,0) and an empty vector shape (0,)
dim = 1000 dim = 1000
A = numpy.zeros((dim, 0), dtype=theano.config.floatX) A = np.zeros((dim, 0), dtype=theano.config.floatX)
b = numpy.zeros((0,), dtype=theano.config.floatX) b = np.zeros((0,), dtype=theano.config.floatX)
tmp = f(A, b) tmp = f(A, b)
assert numpy.allclose(tmp, assert np.allclose(tmp,
numpy.zeros((dim,))) np.zeros((dim,)))
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论