import tensorflow as tf

        )

        # Before and after flux.
        before_flux = batch_win_shaped(band_features["before_flux"])
        after_flux = batch_win_shaped(band_features["after_flux"])
        before_time = batch_win_shaped(band_features["before_time"])
        after_time = batch_win_shaped(band_features["after_time"])

        self.dtime = batch_2win_shaped(
            tf.concat([before_time, after_time], axis=1) - tile_to_2win(closest_time),
        )
        self.dflux = batch_2win_shaped(
            tf.concat([before_flux, after_flux], axis=1) - tile_to_2win(closest_flux),
        )

        # Masking tensor.
        left_mask = _left_mask(
            batch_shaped(
                band_features["before_padding"]),
            window_size)
        right_mask = _right_mask(
            batch_shaped(band_features["after_padding"]), window_size
        )

