提交 1d3531ee authored 作者: Frederic's avatar Frederic

small test speed up.

上级 f9fa66fb
......@@ -292,7 +292,7 @@ GpuAllocTester = makeTester(
def test_shape():
x = GpuArrayType(dtype='float32', broadcastable=[False, False, False])()
v = gpuarray.array(numpy.zeros((3, 4, 5), dtype='float32'))
v = gpuarray.zeros((3, 4, 5), dtype='float32')
f = theano.function([x], x.shape)
topo = f.maker.fgraph.toposort()
assert numpy.all(f(v) == (3, 4, 5))
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论