Loading [Contrib]/a11y/accessibility-menu.js
\u200E
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]