rockpool.nn.losses.base_loss

Loss modules are implemented analogously to losses in PyTorch. All losses inherit from a common Loss class. All loss classes need to implement a forward method which returns a scalar.