o üN iã@sdd„Zdd„ZdS)c Cs’d|||d||d|}|d|d|d|d|d|}||d |d |d|d|}||} |} | | |||gS) a‚Fit coefficients for 4th order polynomial interpolation. Args: y0: function value at the start of the interval. y1: function value at the end of the interval. y_mid: function value at the mid-point of the interval. f0: derivative value at the start of the interval. f1: derivative value at the end of the interval. dt: width of the interval. Returns: List of coefficients `[a, b, c, d, e]` for interpolating with the polynomial `p = a * x ** 4 + b * x ** 3 + c * x ** 2 + d * x + e` for values of `x` between 0 (start of interval) and 1 (end of interval). éééééééé éé ©) Úy0Úy1Úy_midÚf0Úf1ÚdtÚaÚbÚcÚdÚer r úL/home/ubuntu/.local/lib/python3.10/site-packages/torchdiffeq/_impl/interp.pyÚ _interp_fits $,(rcCs„||k||k@sJd |||¡ƒ‚||||}| |dj¡}|d||d}|}|dd…D] }||}|||}q3|S)a·Evaluate polynomial interpolation at the given time point. Args: coefficients: list of Tensor coefficients as created by `interp_fit`. t0: scalar float64 Tensor giving the start of the interval. t1: scalar float64 Tensor giving the end of the interval. t: scalar float64 Tensor giving the desired interpolation point. Returns: Polynomial interpolation of the coefficients at time `t`. z8invalid interpolation, fails `t0 <= t <= t1`: {}, {}, {}éérN)ÚformatÚtoÚdtype)Ú coefficientsÚt0Út1ÚtÚxÚtotalÚx_powerÚ coefficientr r rÚ_interp_evaluates" r&N)rr&r r r rÚs