def layerize(begin_update=None, predict=None, *args, **kwargs):
if begin_update is not None:<EOL><INDENT>return FunctionLayer(begin_update, predict=predict, *args, **kwargs)<EOL><DEDENT>def wrapper(begin_update):<EOL><INDENT>return FunctionLayer(begin_update, *args, **kwargs)<EOL><DEDENT>return wrapper<EOL>
Wrap a function into a layer