import tensorflow as tf

        )

        # 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
        )

        self.mask = batch_2win_shaped(
            tf.logical_and(
                tf.concat([left_mask, right_mask], axis=1), tile_to_2win(self.in_window)
            )
        )

    def dflux_dt(self, clip_magnitude: typing.Optional[float]) -> tf.Tensor:
        """Computes dflux/dt.

        :param clip_magnitude: Option for clipping the magnitude, if dt might be very small.
        :return: <float>[batch_size, 2 * window_size] dflux/dt tensor.
        """
        result = self.dflux / self.dtime
        if clip_magnitude is not None:
            result = tf.clip_by_value(
