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.

build(*args, **kwargs)
Return type

Union[None, Tensor, Iterable[Tensor]]