Downsampled sliding window extractor

downsampling_windows.py

def extract_windows_downsampling(array, clearing_time_index, max_time_steps,
                                 sub_window_size, downsampling_ratio):
    start = clearing_time_index + 1 - (sub_window_size - 1) * downsampling_ratio
    
    K_indices = np.arange(0, sub_window_size * downsampling_ratio,
                          step=downsampling_ratio)
    T_indices = np.arange(0, (max_time_steps + 1) * downsampling_ratio,
                          step=downsampling_ratio)
    sub_windows = np.round(
        start + 
        np.expand_dims(K_indices, 0) +
        np.expand_dims(T_indices, 0).T
    ).astype(np.int)
    
    return array[sub_windows]