раздвижное окно в NumPy

У меня есть бесформенный массив формы (6,2)

[[00,01],
 [10,11],
 [20,21],
 [30,31],
 [40,41],
 [50,51]]

Мне нужно раздвижное окно с размером шага 1 и размером окна 3 нравится это:

[[00,01,10,11,20,21],
 [10,11,20,21,30,31],
 [20,21,30,31,40,41],
 [30,31,40,41,50,51]]

Я ищу тупое решение. Если бы ваше решение могло параметризовать форму исходного массива, а также размер окна и размер шага, это было бы здорово.

Я нашел этот связанный ответИспользование шагов для эффективного фильтра скользящих средних но я не вижу, как указать размер шага там и как свернуть окно из 3d в непрерывный 2d массив. Также этоИтератор с подвижным или скользящим окном в Python но это в Python, и я не уверен, насколько это эффективно. Кроме того, он поддерживает элементы, но не объединяет их в конце концов, если каждый элемент имеет несколько функций.