hxtorch.spiking.modules.hx_module.HXTorchFunctionMixin

class hxtorch.spiking.modules.hx_module.HXTorchFunctionMixin(func: Union[Callable, torch.autograd.function.Function])

Bases: object

__init__(func: Union[Callable, torch.autograd.function.Function]) → None

Parameters

func – Callable function implementing the module’s forward functionality or a torch.autograd.Function implementing the module’s forward and backward operation. TODO: Inform about func args

Methods

__init__(func)	param func Callable function implementing the module’s forward
extra_repr()	Add additional information

Attributes

func

extra_repr() → str

Add additional information

property func