src/transformers/models/reformer/modeling_reformer.py/PositionEmbeddings/forward
class PositionEmbeddings:    def forward(self, position_ids):
        position_embeddings = self.embedding(position_ids)
        position_embeddings = nn.functional.dropout(position_embeddings, p=self.dropout, training=self.training)
        return position_embeddings