src/transformers/models/speecht5/modeling_speecht5.py/SpeechT5RelativePositionalEncoding/__init__
class SpeechT5RelativePositionalEncoding:    def __init__(self, dim, max_length=1000):
        super().__init__()
        self.dim = dim
        self.max_length = max_length
        self.pe_k = torch.nn.Embedding(2 * max_length, dim)