PolynomialConvexScheduler

class flow_matching.path.scheduler.PolynomialConvexScheduler(n: float | int)

Polynomial Scheduler.

kappa_inverse(kappa: Tensor) → Tensor

Computes \(t\) from \(\kappa_t\).

Parameters:

kappa (Tensor) – \(\kappa\), shape (…)

Returns:

t, shape (…)

Return type:

Tensor