mini-ode 0.1.5

A minimalistic ODE solvers library built on top of PyTorch
Documentation
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
use std::fmt;
use tch::{Device, Kind, Tensor};

pub struct MockOptimizer;

impl crate::optimizers::Optimizer for MockOptimizer {
    fn optimize(&self, _f: &dyn Fn(&Tensor) -> Tensor, _x0: &Tensor) -> anyhow::Result<Tensor> {
        Ok(Tensor::zeros(&[1], (Kind::Float, Device::Cpu)))
    }
}

impl fmt::Display for MockOptimizer {
    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
        write!(f, "MockOptimizer")
    }
}