flowmatch_scale_noise

Scale noise for flow matching forward process

Description

Applies the forward process in flow-matching: interpolates between the clean sample and noise.

Usage

flowmatch_scale_noise(sample, timestep, noise, schedule)

Arguments

  • sample: torch tensor. The clean sample.
  • timestep: torch tensor. The current timestep.
  • noise: torch tensor. The noise tensor.
  • schedule: List. The FlowMatch scheduler object.

Value

torch tensor. The noisy sample at timestep t.