@constexpr
def _check_input_3d(input_shape):
    if (len(input_shape) != 3):
        raise ValueError(f'Input should be 3d, but got shape {input_shape}')
@constexpr
def _check_input_3d(input_shape):
    if (len(input_shape) != 3):
        raise ValueError(f'Input should be 3d, but got shape {input_shape}')