openvino.runtime.opset13.multinomial#
- openvino.runtime.opset13.multinomial(probs: Node | int | float | ndarray, num_samples: Node | int | float | ndarray, convert_type: str, with_replacement: bool, log_probs: bool, global_seed: int = 0, op_seed: int = 0, name: str | None = None) Node #
Return a node which generates a sequence of class indices sampled from the multinomial distribution.
- Parameters:
probs – Tensor with probabilities of floating-point type, and shape [batch_size, class_size].
num_samples – Tensor (scalar or 1D) a single element of type i32 or i64, specifying the number of samples to draw from the multinomial distribution.
convert_type – Specifies the output tensor type, possible values: ‘i64’, ‘i32’.
with_replacement – Flag that specifies whether to sample with replacement.
log_probs – Flag that specifies whether probs should be treated as unnormalized log probabilities.
global_seed – Specifies global seed value. Required to be a positive integer or 0.
op_seed – Specifies operational seed value. Required to be a positive integer or 0.
name – The optional new name for output node.
- Returns:
The new node performing Multinomial operation.