tfa.activations.snake

View source on GitHub

Snake activation to learn periodic functions.

https://arxiv.org/abs/2006.08195

logits Input tensor.
frequency A scalar, frequency of the periodic part.

Tensor of the same type and shape as logits.