build_src_rank_and_local_expert_id

paddle.incubate.nn.functional. build_src_rank_and_local_expert_id ( expert_num_global_tensor: Tensor, expert_num_global: list, num_local_experts: int, name: str | None = None ) Tensor [source]
Parameters
  • expert_num_global_tensor

  • expert_num_global

  • num_local_experts

Returns: