fc84b744240b6d8e90ba725ae0819af585badafc,layer/downsample.py,DownSampleLayer,layer_op,#DownSampleLayer#Any#,28

Before Change

    def layer_op(self, input_tensor):
        spatial_rank = input_tensor.get_shape().ndims - 2
        assert(spatial_rank > 0)
        output_tensor = tf.nn.pool(
                input=input_tensor,
                window_shape=[self.kernel_size] * spatial_rank,
                pooling_type=self.func,

After Change

        super(DownSampleLayer, self).__init__(name=self.layer_name)

    def layer_op(self, input_tensor):
        spatial_rank = layer_util.infer_spatial_rank(input_tensor)
        output_tensor = tf.nn.pool(
                input=input_tensor,
                window_shape=[self.kernel_size] * spatial_rank,
                pooling_type=self.func,

Instances