2b189592e0a169fc991bec98f39bbe8405e3440b,torchdiffeq/_impl/fixed_grid.py,Midpoint,_step_func,#Midpoint#Any#Any#Any#Any#,15

Before Change

    order = 2

    def _step_func(self, func, t, dt, y):
        y_mid = tuple(y_ + f_ * dt / 2 for y_, f_ in zip(y, func(t + self.eps, y)))
        return tuple(dt * f_ for f_ in func(t + dt / 2, y_mid))


class RK4(FixedGridODESolver):
    order = 4

After Change

    order = 2

    def _step_func(self, func, t, dt, y):
        half_dt = 0.5 * dt
        y_mid = y + func(t + self.eps, y) * half_dt
        return dt * func(t + half_dt, y_mid)


class RK4(FixedGridODESolver):
    order = 4

Instances