gensbi.flow_matching package# Subpackages# gensbi.flow_matching.loss package Submodules gensbi.flow_matching.loss.continuous_loss module ContinuousFMLoss Module contents ContinuousFMLoss gensbi.flow_matching.path package Subpackages gensbi.flow_matching.path.scheduler package Submodules gensbi.flow_matching.path.scheduler.schedule_transform module gensbi.flow_matching.path.scheduler.scheduler module Module contents Submodules gensbi.flow_matching.path.affine module AffineProbPath AffineProbPath.epsilon_to_target() AffineProbPath.epsilon_to_velocity() AffineProbPath.sample() AffineProbPath.target_to_epsilon() AffineProbPath.target_to_velocity() AffineProbPath.velocity_to_epsilon() AffineProbPath.velocity_to_target() CondOTProbPath gensbi.flow_matching.path.path module ProbPath ProbPath.assert_sample_shape() ProbPath.sample() gensbi.flow_matching.path.path_sample module DiscretePathSample DiscretePathSample.x_1 DiscretePathSample.x_0 DiscretePathSample.t DiscretePathSample.x_t DiscretePathSample.t DiscretePathSample.x_0 DiscretePathSample.x_1 DiscretePathSample.x_t PathSample PathSample.x_1 PathSample.x_0 PathSample.t PathSample.x_t PathSample.dx_t PathSample.dx_t PathSample.t PathSample.x_0 PathSample.x_1 PathSample.x_t Module contents AffineProbPath AffineProbPath.epsilon_to_target() AffineProbPath.epsilon_to_velocity() AffineProbPath.sample() AffineProbPath.target_to_epsilon() AffineProbPath.target_to_velocity() AffineProbPath.velocity_to_epsilon() AffineProbPath.velocity_to_target() CondOTProbPath PathSample PathSample.x_1 PathSample.x_0 PathSample.t PathSample.x_t PathSample.dx_t PathSample.dx_t PathSample.t PathSample.x_0 PathSample.x_1 PathSample.x_t ProbPath ProbPath.assert_sample_shape() ProbPath.sample() gensbi.flow_matching.solver package Submodules gensbi.flow_matching.solver.ode_solver module ODESolver ODESolver.get_sampler() ODESolver.get_unnormalized_logprob() ODESolver.sample() ODESolver.unnormalized_logprob() gensbi.flow_matching.solver.sde_solver module BaseSDESolver BaseSDESolver.get_f_tilde() BaseSDESolver.get_g_tilde() BaseSDESolver.get_sampler() BaseSDESolver.get_score() BaseSDESolver.sample() NonSingular NonSingular.get_f_tilde() NonSingular.get_g_tilde() ZeroEnds ZeroEnds.get_f_tilde() ZeroEnds.get_g_tilde() gensbi.flow_matching.solver.solver module Solver Solver.sample() gensbi.flow_matching.solver.utils module get_nearest_times() Module contents NonSingular NonSingular.get_f_tilde() NonSingular.get_g_tilde() ODESolver ODESolver.get_sampler() ODESolver.get_unnormalized_logprob() ODESolver.sample() ODESolver.unnormalized_logprob() Solver Solver.sample() ZeroEnds ZeroEnds.get_f_tilde() ZeroEnds.get_g_tilde() gensbi.flow_matching.utils package Subpackages gensbi.flow_matching.utils.manifolds_not_implemented package Submodules gensbi.flow_matching.utils.manifolds_not_implemented.manifold module gensbi.flow_matching.utils.manifolds_not_implemented.sphere module gensbi.flow_matching.utils.manifolds_not_implemented.torus module gensbi.flow_matching.utils.manifolds_not_implemented.utils module Module contents Submodules gensbi.flow_matching.utils.utils module expand_tensor_like() unsqueeze_to_match() Module contents expand_tensor_like() unsqueeze_to_match() Module contents#