//clika-runtime/io.clika.runtime/Ops/multinomial

multinomial

[common]
fun multinomial(probabilities: Tensor, numSamples: Long, replacement: Boolean = false, device: Placement? = null): Tensor

multinomial(probabilities: Tensor, numSamples: Long, replacement: Boolean = false, device: Placement? = null): the multinomial operator. Categorical sampling: draws num_samples category indices per row of a weight tensor. probabilities holds non-negative weights (they need not sum to 1), typically [batch, num_categories]; the output replaces the category axis with num_samples and is Int64. Without replacement each row samples distinct categories.