def expand(arr, axis, length):
    new_shape = list(arr.shape)
new_shape.insert(axis, length)
new_strides = list(arr.strides)
new_strides.insert(axis, 0)
return np.lib.stride_tricks.as_strided(arr, new_shape, new_strides)
Расширение массива NumPy по дополнительному измерению