提交 6dd05db4 authored 作者: Gijs van Tulder's avatar Gijs van Tulder

Use new built-in T.tensor6 in tests.

上级 61b4d95c
...@@ -1566,10 +1566,8 @@ def test_dnn_batchnorm_train(): ...@@ -1566,10 +1566,8 @@ def test_dnn_batchnorm_train():
raise SkipTest(dnn.dnn_available.msg) raise SkipTest(dnn.dnn_available.msg)
utt.seed_rng() utt.seed_rng()
tensor6 = T.TensorType(theano.config.floatX, (False,) * 6)
for mode in ('per-activation', 'spatial'): for mode in ('per-activation', 'spatial'):
for vartype in (tensor6, T.tensor5, T.tensor4, T.tensor3, T.matrix, T.vector): for vartype in (T.tensor6, T.tensor5, T.tensor4, T.tensor3, T.matrix, T.vector):
x, scale, bias, running_mean, running_var = (vartype(n) x, scale, bias, running_mean, running_var = (vartype(n)
for n in ('x', 'scale', 'bias', for n in ('x', 'scale', 'bias',
'running_mean', 'running_mean',
...@@ -1804,10 +1802,8 @@ def test_batchnorm_inference(): ...@@ -1804,10 +1802,8 @@ def test_batchnorm_inference():
raise SkipTest(dnn.dnn_available.msg) raise SkipTest(dnn.dnn_available.msg)
utt.seed_rng() utt.seed_rng()
tensor6 = T.TensorType(theano.config.floatX, (False,) * 6)
for mode in ('per-activation', 'spatial'): for mode in ('per-activation', 'spatial'):
for vartype in (tensor6, T.tensor5, T.tensor4, T.tensor3, T.matrix, T.vector): for vartype in (T.tensor6, T.tensor5, T.tensor4, T.tensor3, T.matrix, T.vector):
x, scale, bias, mean, var = (vartype(n) x, scale, bias, mean, var = (vartype(n)
for n in ('x', 'scale', 'bias', 'mean', 'var')) for n in ('x', 'scale', 'bias', 'mean', 'var'))
ndim = x.ndim ndim = x.ndim
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论