Module: tfa.layers.snake

View source on GitHub

Implements Snake layer.

Classes

class Snake: Snake layer to learn periodic functions with the trainable frequency scalar.