Method diffusion_fn

diff2flow/flow.py:261–263

(x, t)

    ):

        def diffusion_fn(x, t):
            diffusion = self.schedule.compute_diffusion(x, t, form=diffusion_form, norm=diffusion_norm)
            return diffusion
        
        sde_drift = \
            lambda x, t, model, **kwargs: \