src/transformers/models/transfo_xl/modeling_tf_transfo_xl.py/TFPositionalEmbedding/__init__
class TFPositionalEmbedding:    def __init__(self, demb, **kwargs):
        super().__init__(**kwargs)

        self.inv_freq = 1 / (10000 ** (tf.range(0, demb, 2.0) / demb))