Function test_guidance

tests/unittests/test_diffusion_math.cpp:32–59

}

void test_guidance() {
    const std::vector<float> pred_cond{0.4F, -0.2F, 0.8F, -0.5F, 0.1F, 0.6F};
    const std::vector<float> pred_uncond{-0.1F, 0.3F, 0.2F, 0.4F, -0.6F, 0.0F};
    require_vector_close(
        engine::sampling::cfg_guidance(pred_cond, pred_uncond, 2.5F),
        {1.15F, -0.95F, 1.7F, -1.85F, 1.15F, 1.5F},
        2.0e-6F,
        "cfg python reference");

    std::vector<float> momentum(6, 0.0F);
    require_vector_close(
        engine::sampling::apg_guidance(pred_cond, pred_uncond, 2.0F, 2, 3, momentum),
        {0.265853659F, -0.02F, 0.728F, -0.60731709F, 0.46F, 0.696F},
        2.0e-6F,
        "apg python reference");

    const std::vector<float> latents{1.2F, -0.7F, 0.3F, 2.0F, -1.1F, 0.5F};
    require_vector_close(
        engine::sampling::adg_guidance(latents, pred_cond, pred_uncond, 0.6F, 2.25F, 2, 3),
        {0.35072118F, -0.34340942F, 1.41628838F, -0.47153577F, 0.47726715F, 1.64375424F},
        2.0e-6F,
        "adg python reference");

    const std::vector<float> zero_latents{0.0F, 0.0F};
    const std::vector<float> zero_pred{0.0F, 0.0F};
    const auto zero_adg = engine::sampling::adg_guidance(zero_latents, zero_pred, zero_pred, 1.0F, 2.0F, 1, 2);
    require_vector_close(zero_adg, {0.0F, 0.0F}, 1.0e-6F, "adg zero norm finite");
}

void test_velocity_and_steps() {
    std::vector<float> velocity{3.0F, 4.0F};