In-place patching (framing of 2d inputs)

frame2d.py

def frame2d(x, patch_length, hop_length):
    
    n_frames = 1 + (x.shape[-1] - patch_length) // hop_length
    
    x_frame = np.lib.stride_tricks.as_strided(x, shape=(x.shape[0], patch_length, n_frames),
                                                    strides=(1 * x.itemsize, x.shape[0] * x.itemsize, hop_length * x.shape[0] * x.itemsize))
    return x_frame