code: def _segment_relative_position_bucket(
      self, query_segment, key_segment
  ):
      return query_segment * self.num_segments + key_segment
  
file_path: DeepKE/example/llm/CPM-Bee/src/cpm_live/layers/position_embedding.py
imports: import math
  from typing import Union
  import torch
  import bmtrain
  import torch.nn.functional
label_desc: Use image node_id 3 for calling the SegmentPositionEmbedding obj's underlying member method code with example usage: obj._segment_relative_position_bucket(query_segment, key_segment) and returns: unknown