863a9dcee691e850e79d97a16abb977e24324fa1,tests/downsample_test.py,DownSampleTest,test_2d_const_shape,#DownSampleTest#,68

Before Change

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

    def test_2d_const_shape(self):
        x = self.get_2d_input()
        down_sample_layer = DownSampleLayer("CONSTANT", 3, 3)
        out_down_sample_const = 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_const)
            self.assertAllClose((2, 6, 6, 8), out.shape)


if __name__ == "__main__":
    tf.test.main()

After Change

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

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


if __name__ == "__main__":
    tf.test.main()

Instances