MolecularDiffusion.modules.models.syncogen.diffusion.interpolation.linear¶
Linear interpolator.
Classes¶
Linear interpolator for flow matching. |
Module Contents¶
- class MolecularDiffusion.modules.models.syncogen.diffusion.interpolation.linear.LinearInterpolator¶
Bases:
MolecularDiffusion.modules.models.syncogen.diffusion.interpolation.base.InterpolatorBaseLinear interpolator for flow matching.
Implements simple linear interpolation: φ_t(C0, C1) = (1-t) * C0 + t * C1
- interpolate(C0: torch.Tensor, C1: torch.Tensor, t: torch.Tensor) torch.Tensor¶
Linearly interpolate between C0 and C1.
- Parameters:
C0 – Initial coordinates tensor
C1 – Final coordinates tensor
t – Time tensor, shape [B] or [B, 1, …]
- Returns:
Linearly interpolated coordinates