863a9dcee691e850e79d97a16abb977e24324fa1,tests/downsample_test.py,DownSampleTest,test_2d_max_shape,#DownSampleTest#,48

Before Change

            self.assertAllClose((2, 6, 6, 6, 8), out.shape)

    def test_2d_max_shape(self):
        x = self.get_2d_input()
        down_sample_layer = DownSampleLayer("MAX", 3, 3)
        out_down_sample_max = down_sample_layer(x)
        print(down_sample_layer)
        with self.test_session() as sess:
            sess.run(tf.global_variables_initializer())
            out = sess.run(out_down_sample_max)
            self.assertAllClose((2, 6, 6, 8), out.shape)

    def test_2d_avg_shape(self):
        x = self.get_2d_input()
        down_sample_layer = DownSampleLayer("AVG", 2, 2)
        out_down_sample_avg = down_sample_layer(x)

After Change

                                              output_shape=(2, 6, 8, 8, 8))

    def test_2d_max_shape(self):
        input_param = {"func": "CONSTANT",
                       "kernel_size": [1, 3],
                       "stride": 3}
        self._test_nd_downsample_output_shape(rank=2,
                                              param_dict=input_param,
                                              output_shape=(2, 6, 6, 8))

    def test_2d_avg_shape(self):
        input_param = {"func": "AVG",
                       "kernel_size": [2, 3],

Instances