Add an "input args" parameter when defining derivative actions.

Add an "input args" parameter when defining derivative actions. #596