Skip to main content

NeuralOdeSolver

oxiphysics_core::neural_ode

Struct NeuralOdeSolver 

Source 
pub struct NeuralOdeSolver {
    pub func: NeuralOdeFunc,
    pub rtol: f64,
    pub atol: f64,
}
Expand description

Integrates a NeuralOdeFunc from t0 to t1 using fixed-step RK4 or adaptive DOPRI5.

Fields§

§func: NeuralOdeFunc

The parameterised ODE dynamics.

§rtol: f64

Relative tolerance for adaptive step control.

§atol: f64

Absolute tolerance for adaptive step control.

Implementations§

Source§

impl NeuralOdeSolver

Source

pub fn new(func: NeuralOdeFunc, rtol: f64, atol: f64) -> Self

Create a new solver wrapping func with the given tolerances.

Source

pub fn solve_rk4(&self, z0: &[f64], t0: f64, t1: f64, dt: f64) -> Vec<f64>

Solve from z0 at t0 to t1 using fixed-step RK4 with step size dt.

Returns the final state at t1.

Source

pub fn solve_dopri5( &self, z0: &[f64], t0: f64, t1: f64, dt_init: f64, ) -> Vec<f64>

Solve from z0 to t1 using adaptive DOPRI5.

Returns the final state at t1.

Source

pub fn solve_rk4_trajectory( &self, z0: &[f64], ts: &[f64], dt: f64, ) -> Vec<Vec<f64>>

Return all intermediate states at times ts using RK4.

ts must be sorted ascending and the first element is ignored (treated as t0). The returned vector has one entry per element of ts.

Trait Implementations§

Source§

impl Clone for NeuralOdeSolver

Source§

fn clone(&self) -> NeuralOdeSolver

Returns a duplicate of the value. Read more
1.0.0 (const: unstable) · Source§

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source. Read more
Source§

impl Debug for NeuralOdeSolver

Source§

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter. Read more

Auto Trait Implementations§

Blanket Implementations§

Source§

impl<T> Any for T
where T: 'static + ?Sized,

Source§

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more
Source§

impl<T> Borrow<T> for T
where T: ?Sized,

Source§

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more
Source§

impl<T> BorrowMut<T> for T
where T: ?Sized,

Source§

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more
Source§

impl<T> CloneToUninit for T
where T: Clone,

Source§

unsafe fn clone_to_uninit(&self, dest: *mut u8)

🔬This is a nightly-only experimental API. (clone_to_uninit)
Performs copy-assignment from self to dest. Read more
Source§

impl<T> From<T> for T

Source§

fn from(t: T) -> T

Returns the argument unchanged.

Source§

impl<T, U> Into<U> for T
where U: From<T>,

Source§

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

Source§

impl<T> Same for T

Source§

type Output = T

Should always be Self
Source§

impl<SS, SP> SupersetOf<SS> for SP
where SS: SubsetOf<SP>,

Source§

fn to_subset(&self) -> Option<SS>

The inverse inclusion map: attempts to construct self from the equivalent element of its superset. Read more
Source§

fn is_in_subset(&self) -> bool

Checks if self is actually part of its subset T (and can be converted to it).
Source§

fn to_subset_unchecked(&self) -> SS

Use with care! Same as self.to_subset but without any property checks. Always succeeds.
Source§

fn from_subset(element: &SS) -> SP

The inclusion map: converts self to the equivalent element of its superset.
Source§

impl<T> ToOwned for T
where T: Clone,

Source§

type Owned = T

The resulting type after obtaining ownership.
Source§

fn to_owned(&self) -> T

Creates owned data from borrowed data, usually by cloning. Read more
Source§

fn clone_into(&self, target: &mut T)

Uses borrowed data to replace owned data, usually by cloning. Read more
Source§

impl<T, U> TryFrom<U> for T
where U: Into<T>,

Source§

type Error = Infallible

The type returned in the event of a conversion error.
Source§

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.
Source§

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

Source§

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.
Source§

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.