Enhance layer_function_generator to accept *args

[reyoung]

In current fluid API, it is noising to use a layer which is warped by layer_function_generator. For example, we must use mean(x=blah) to create a mean operator. The x= cannot be ignored. We should make these functions able to accept *args so that we can write mean(blah).