def __exit__(self, *args):
        if self.dilation_factor > 1:
            self._tensor = tf.batch_to_space_nd(self._tensor,
                                                self.block_shape,
                                                self.zero_paddings,
                                                name='de-dilate')