norse.torch.functional.lift

norse.torch.functional.lift(activation, p=None)

Creates a lifted version of the given activation function which applies the activation function in the temporal domain. The returned callable can be applied later as if it was a regular activation function, but the input is now assumed to be a tensor whose first dimension is time.

Parameters:

activation (Callable[[torch.Tensor, Any, Any], Tuple[torch.Tensor, Any]]):

The activation function that takes an input tensor, an optional state, and an optional parameter object and returns a tuple of (spiking output, neuron state). The returned spiking output includes the time domain.

p (Any): An optional parameter object to hand to the activation function.

Returns:

A :class:`.Callable`_ that, when applied, evaluates the activation function N times, where N is the size of the outer (temporal) dimension. The application will provide a tensor of shape (time, …).