spd_learn.models.PhaseDelay

class spd_learn.models.PhaseDelay(order: int = 1, lag: int = 1, device=None, dtype=None)

Bases: Module

Phase-Space Embedding Layer.

This class implements the phase-space embedding layer [Carrara* et al., 2024].

Parameters:

• order (int, default=1) – Order of the phase space.

• lag (int, default=1) – Delay of the phase space.

forward(input: Tensor) → Tensor

Performs lagged concatenation on the input tensor.

Parameters:

input (torch.Tensor) – Input tensor of shape (batch, channels, time_length).

Returns:

Augmented tensor of shape (batch, channels * order, time_out), where time_out = n_times - (order * lag).

Return type:

torch.Tensor