Method time_schedule

cake-core/src/models/luxtts/euler_solver.rs:25–37

Generate the time schedule: linspace(0, 1, num_steps+1) with t_shift applied.

(&self)

    /// Generate the time schedule: linspace(0, 1, num_steps+1) with t_shift applied.
    pub fn time_schedule(&self) -> Vec<f32> {
        let mut times = Vec::with_capacity(self.num_steps + 1);
        for i in 0..=self.num_steps {
            let t = i as f32 / self.num_steps as f32;
            let t_shifted = if (self.t_shift - 1.0).abs() > 1e-6 {
                self.t_shift * t / (1.0 + (self.t_shift - 1.0) * t)
            } else {
                t
            };
            times.push(t_shifted);
        }
        times
    }

    /// Perform one flow matching Euler step.
    /// Returns updated x for the next timestep.