files/data_io/transforms/spike_augmentation.py


1
2
3
4
5
6
7
8
9
10

import torch

class SpikeJitter:
    """Add temporal jitter noise to spikes."""
    def __init__(self, std=0.01):
        self.std = std

    def __call__(self, spikes: torch.Tensor) -> torch.Tensor:
        # TODO: add random jitter to spike timings
        return spikes