def METHOD_NAME():
    shape = (10, 10)
    x = relay.var("x", shape=shape)
    y = relay.expand_dims(x, axis=0)
    func = relay.Function([x], y)
    _construct_model(func)
    y = relay.expand_dims(x, axis=-1)
    func = relay.Function([x], y)
    _construct_model(func)