This will be useful for SNNs, as the relationship between spiking
See also pytorch operator
Fixes #173.