moe_permute

paddle.nn.functional. moe_permute ( hidden_states: Tensor, scale: Tensor | None, expert_routemap_topk: Tensor, expert_prob_topk: Tensor, num_experts: int, tokens_per_expert: list, padding_alignment: int, name: str | None = None ) tuple[Tensor, Tensor, Tensor, Tensor] [source]