提交 10f1311e authored 作者: Seon-Wook Park's avatar Seon-Wook Park

Remove infer_shape tests for compress

上级 da637d50
...@@ -382,18 +382,6 @@ class CompressTester(utt.InferShapeTester): ...@@ -382,18 +382,6 @@ class CompressTester(utt.InferShapeTester):
assert tested.shape == expected.shape assert tested.shape == expected.shape
assert numpy.allclose(tested, expected) assert numpy.allclose(tested, expected)
def test_infer_shape(self):
for axis, cond, shape in zip(self.axis_list, self.cond_list, self.shape_list):
cond_var = theano.tensor.ivector()
data = numpy.random.random(size=shape).astype(theano.config.floatX)
data_var = theano.tensor.matrix()
self._compile_and_check([cond_var, data_var],
[self.op(cond_var, data_var, axis=axis)],
[cond, data],
tensor.AdvancedSubtensor1,
warn=False)
class TestRepeatOp(utt.InferShapeTester): class TestRepeatOp(utt.InferShapeTester):
def _possible_axis(self, ndim): def _possible_axis(self, ndim):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论