moe_gate_dispatch

paddle.incubate.nn.functional. moe_gate_dispatch ( x: Tensor, gate_logits: Tensor, corr_bias: Tensor, k: int, capacity: int, use_pad: bool, name: str | None = None ) → Tensor [source]

Parameters

• x –

• gate_logits –

• corr_bias –

• k –

• capacity –

• use_pad –

Returns

combine_weights: scatter_index: expert_offset: expert_id:

Return type

y