hxtorch.spiking.functional.mock.discrete

Define torch.autograd.Function to mock discretization (e.g. weight discretization).

Classes

Discrete(*args, **kwargs)

Discretize values of a tensor.