scimba_torch.physical_models.ode.damped_harmonic_oscillator

Implementation of a damped harmonic oscillator ODE.

Classes

DampedHarmonicOscillator(space, init, **kwargs)

Implementation of a damped harmonic oscillator ODE.

class DampedHarmonicOscillator(space, init, **kwargs)

Bases: AbstractODE

Implementation of a damped harmonic oscillator ODE.

\[\begin{split}\frac{dx}{dt} = - \mu x - y, \\ \frac{dy}{dt} = x - \mu y, \\\end{split}\]

with an initial condition given by the function init.

Parameters:

• space (AbstractApproxSpace) – The approximation space for the problem

• init (Callable) – Callable for the initial condition

• **kwargs – Additional keyword arguments

time_operator(w, t, mu)

Apply the ODE operator.

Parameters:

• w (MultiLabelTensor) – Solution tensor

• t (LabelTensor) – Temporal coordinate tensor

• mu (LabelTensor) – Parameter tensor

Return type:

Tensor | tuple[Tensor, ...]

Returns:

Operator tensor

functional_operator(func, t, mu, theta)

Compute the functional operator.

Parameters:

• func (VarArgCallable) – Callable representing the function

• t (Tensor) – Temporal coordinate tensor

• mu (Tensor) – Parameter tensor

• theta (Tensor) – Additional parameters tensor

Return type:

Tensor

Returns:

Functional operator tensor