myoverse.transforms.RMS._compute_window#

RMS._compute_window(x_unfolded)[source]#

Compute the window-wise statistic.

Parameters:

x_unfolded (torch.Tensor) – Unfolded tensor with windows in the last dimension. Shape: (…, n_windows, window_size)

Returns:

Reduced tensor. Shape: (…, n_windows)

Return type:

torch.Tensor