def METHOD_NAME(inp_shape):
    """ test Flatten default parameters """
    batch_size = 2
    inp = torch.zeros((batch_size, ) + inp_shape)
    out = Flatten()(inp)
    assert out.ndim == 2
    assert out.shape == (inp.shape[0], np.prod(inp_shape))