Update docstring.
Use setup_vars() method to define variables.
Expose layer preactivations via the .preacts attribute.