moe_gate_dispatch_partial_nosoftmaxtopk

paddle.incubate.nn.functional. moe_gate_dispatch_partial_nosoftmaxtopk ( x: Tensor, combine_weights: Tensor, expert_id: Tensor, k: int, capacity: int, num_experts: int, use_pad: bool, expert_start_index: int, expert_end_index: int, reverse_token_drop: bool, name: str | None = None ) → tuple[Tensor, Tensor, Tensor, Tensor, Tensor, Tensor] [source]