提交 62c86dcb authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Test on smaller tensors

上级 c63d4f3c
import unittest import unittest
import numpy import numpy
import copy import copy
import itertools
import theano import theano
import theano.tensor as T import theano.tensor as T
...@@ -30,12 +31,13 @@ class TestConv2d(unittest.TestCase): ...@@ -30,12 +31,13 @@ class TestConv2d(unittest.TestCase):
def setUp(self): def setUp(self):
super(TestConv2d, self).setUp() super(TestConv2d, self).setUp()
self.inputs_shapes = [(16, 1, 12, 12), (16, 1, 18, 18), (2, 1, 24, 24), self.inputs_shapes = [(8, 1, 12, 12), (8, 1, 18, 18), (2, 1, 4, 4),
(6, 1, 20, 20), (2, 1, 32, 20), (1, 5, 32, 32)] (6, 1, 10, 11), (2, 1, 6, 5), (1, 5, 9, 9)]
self.filters_shapes = [(10, 1, 2, 2), (10, 1, 3, 3), (10, 1, 2, 2), self.filters_shapes = [(5, 1, 2, 2), (4, 1, 3, 3), (2, 1, 3, 3),
(1, 1, 2, 5), (5, 1, 2, 2), (15, 5, 2, 2)] (1, 1, 2, 5), (4, 1, 2, 2), (4, 5, 2, 2)]
self.subsamples = [(1, 1), (2, 2), (2, 4)] self.subsamples = [(1, 1), (2, 2), (2, 4)]
self.border_modes = ["valid", "full", (0, 0), (1, 1), (5, 5), (5, 2)] self.border_modes = ["valid", "full", (0, 0), (1, 1), (5, 5), (5, 2)]
self.filters_flip = [True, False]
def get_output_shape(self, inputs_shape, filters_shape, subsample, border_mode): def get_output_shape(self, inputs_shape, filters_shape, subsample, border_mode):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论