Skip to content

P
pytensor
提交 ff8f57f1 authored 作者: abergeron's avatar abergeron
浏览文件
操作

Merge pull request #1858 from nouiz/faster_test

Faster test
上级 eb4e7715 a52e4b45
隐藏空白字符变更
内嵌 并排
正在显示 包含 53 行增加 和 51 行删除
theano/tensor/tests/test_keepdims.py
浏览文件 @ ff8f57f1
from nose.plugins.attrib import attr import unittest
from nose.plugins.attrib import attr
import numpy import numpy
import theano
from theano import tensor, function from theano import tensor, function
import unittest
# this tests other ops to ensure they keep the dimensions of their # this tests other ops to ensure they keep the dimensions of their
# inputs correctly # inputs correctly
class TestKeepDims(unittest.TestCase): class TestKeepDims(unittest.TestCase):
def makeKeepDims_local(self, x, y, axis): def makeKeepDims_local(self, x, y, axis):
x = tensor.as_tensor_variable(x)
y = tensor.as_tensor_variable(y)
if axis is None: if axis is None:
axis = numpy.arange(x.ndim) newaxis = range(x.ndim)
elif isinstance(axis, int): elif isinstance(axis, int):
axis = [axis] if axis < 0:
newaxis = [axis + x.type.ndim]
else:
newaxis = [axis]
else:
newaxis = []
for a in axis:
if a < 0:
a += x.type.ndim
newaxis.append(a)
i = 0 i = 0
newaxis = []
for a in axis:
if a < 0:
a += x.type.ndim
newaxis.append(a)
new_dims = [] new_dims = []
for j, _ in enumerate(x.shape): for j, _ in enumerate(x.shape):
if j in newaxis: if j in newaxis:
...@@ -38,6 +41,9 @@ class TestKeepDims(unittest.TestCase): ...@@ -38,6 +41,9 @@ class TestKeepDims(unittest.TestCase):
x = tensor.dtensor3() x = tensor.dtensor3()
a = numpy.random.rand(3, 2, 4) a = numpy.random.rand(3, 2, 4)
# We don't need to test all opt and C code, as this is tested
# by the ops tests.
mode = theano.compile.Mode(optimizer="fast_compile", linker="py")
# 'max_and_argmax' has two outputs and can be specified with either # 'max_and_argmax' has two outputs and can be specified with either
# a single or every axis: # a single or every axis:
...@@ -46,19 +52,23 @@ class TestKeepDims(unittest.TestCase): ...@@ -46,19 +52,23 @@ class TestKeepDims(unittest.TestCase):
[-2, -3, 2]]: [-2, -3, 2]]:
op = tensor.max_and_argmax op = tensor.max_and_argmax
keep_param = function([x], op(x, axis=axis, keepdims=True)[0]) f = function([x], [op(x, axis=axis, keepdims=True)[0],
keep_synth = function([x], self.makeKeepDims_local(x, self.makeKeepDims_local(
op(x, axis=axis, keepdims=False)[0], axis)) x, op(x, axis=axis, keepdims=False)[0],
axis)],
assert numpy.allclose(keep_param(a), keep_synth(a)) mode=mode)
assert keep_param(a).shape == keep_synth(a).shape ans1, ans2 = f(a)
assert numpy.allclose(ans1, ans2)
keep_param = function([x], op(x, axis=axis, keepdims=True)[1]) assert ans1.shape == ans2.shape
keep_synth = function([x], self.makeKeepDims_local(x,
op(x, axis=axis, keepdims=False)[1], axis)) f = function([x], [op(x, axis=axis, keepdims=True)[1],
self.makeKeepDims_local(
assert numpy.allclose(keep_param(a), keep_synth(a)) x, op(x, axis=axis, keepdims=False)[1],
assert keep_param(a).shape == keep_synth(a).shape axis)],
mode=mode)
ans1, ans2 = f(a)
assert numpy.allclose(ans1, ans2)
assert ans1.shape == ans2.shape
# the following ops can be specified with either a single axis or every # the following ops can be specified with either a single axis or every
# axis: # axis:
...@@ -66,38 +76,30 @@ class TestKeepDims(unittest.TestCase): ...@@ -66,38 +76,30 @@ class TestKeepDims(unittest.TestCase):
for axis in [0, 1, 2, [0], [1], [2], None, [0, 1, 2], for axis in [0, 1, 2, [0], [1], [2], None, [0, 1, 2],
[-1], [-2], [-3], [-1, -2, -3], [0, -2, 2]]: [-1], [-2], [-3], [-1, -2, -3], [0, -2, 2]]:
keep_param = function([x], op(x, axis=axis, keepdims=True)) f = function([x], [op(x, axis=axis, keepdims=True),
keep_synth = function([x], self.makeKeepDims_local(x, self.makeKeepDims_local(
op(x, axis=axis, keepdims=False), axis)) x, op(x, axis=axis, keepdims=False),
axis)],
assert numpy.allclose(keep_param(a), keep_synth(a)) mode=mode)
assert keep_param(a).shape == keep_synth(a).shape ans1, ans2 = f(a)
assert numpy.allclose(ans1, ans2)
keep_param = function([x], op(x, axis=None, keepdims=True)) assert ans1.shape == ans2.shape
keep_synth = function([x], self.makeKeepDims_local(x,
op(x, axis=None, keepdims=False), None))
assert numpy.allclose(keep_param(a), keep_synth(a))
assert keep_param(a).shape == keep_synth(a).shape
# the following ops can be specified with a freely specified axis # the following ops can be specified with a freely specified axis
# parameter # parameter
for op in ([tensor.sum, tensor.prod, tensor.mean, tensor.var, for op in ([tensor.sum, tensor.prod, tensor.mean, tensor.var,
tensor.std, tensor.all, tensor.any, tensor.std, tensor.all, tensor.any,
tensor.max, tensor.min]): tensor.max, tensor.min]):
for axis in [0, 1, 2, [0], [1], [2], [0, 1], [1, 2], [0, 1, 2], for axis in [0, 1, 2, [0], [1], [2], None,
[0, 1], [1, 2], [0, 1, 2],
[-1], [-2], [-3], [-1, -2], [-1, -2, -3], [0, -2, 2]]: [-1], [-2], [-3], [-1, -2], [-1, -2, -3], [0, -2, 2]]:
keep_param = function([x], op(x, axis=axis, keepdims=True)) f = function([x], [op(x, axis=axis, keepdims=True),
keep_synth = function([x], self.makeKeepDims_local(x, self.makeKeepDims_local(
op(x, axis=axis, keepdims=False), axis)) x, op(x, axis=axis, keepdims=False),
axis)],
assert numpy.allclose(keep_param(a), keep_synth(a)) mode=mode)
assert keep_param(a).shape == keep_synth(a).shape
keep_param = function([x], op(x, axis=None, keepdims=True))
keep_synth = function([x], self.makeKeepDims_local(x,
op(x, axis=None, keepdims=False), None))
assert numpy.allclose(keep_param(a), keep_synth(a)) ans1, ans2 = f(a)
assert keep_param(a).shape == keep_synth(a).shape assert numpy.allclose(ans1, ans2)
assert ans1.shape == ans2.shape
Markdown 格式
0% 或
您添加了 0 人 到此讨论。请谨慎行事。
请先完成此评论的编辑！
请 注册 或者 后发表评论