src/transformers/models/nezha/modeling_nezha.py/NezhaRelativePositionsEncoding/forward
class NezhaRelativePositionsEncoding:    def forward(self, length):
        return self.positions_encoding[:length, :length, :]