19.5. Modules
- class popxl.Module
Callable class that user-defined layers can inherit from.
The
build
method should be overridden with a function to build the subgraph.The benefit of inheriting from this class rather than passing a function is that you can save input tensors as fields on
self
then later, when you call the subgraph, you can pass a mapping from the input tensor IDs to the corresponding parent tensor you wish to pass.