rockpool.nn.modules.torch.torch_module

Provide a base class for build Torch-compatible modules

Functions

to_float_tensor(x)

Classes

TorchModule(*args, **kwargs)

Base class for modules that are compatible with both Torch and Rockpool

TorchModuleParameters

A dict subclass that supports conversion to raw values