|
|
|
@ -401,7 +401,7 @@ class RandomChoiceWithMask(PrimitiveWithInfer):
|
|
|
|
|
|
|
|
|
|
Inputs:
|
|
|
|
|
- **input_x** (Tensor[bool]) - The input tensor.
|
|
|
|
|
The input tensor rank must be greater than or equal to 1 and less than or equal to 5.
|
|
|
|
|
The input tensor rank must be greater than or equal to 1 and less than or equal to 5.
|
|
|
|
|
|
|
|
|
|
Outputs:
|
|
|
|
|
Two tensors, the first one is the index tensor and the other one is the mask tensor.
|
|
|
|
@ -530,7 +530,7 @@ class Multinomial(PrimitiveWithInfer):
|
|
|
|
|
seed2 (int): Random seed2, must be non-negative. Default: 0.
|
|
|
|
|
Inputs:
|
|
|
|
|
- **input** (Tensor[float32]) - the input tensor containing the cumsum of probabilities, must be 1 or 2
|
|
|
|
|
dimensions.
|
|
|
|
|
dimensions.
|
|
|
|
|
- **num_samples** (int32) - number of samples to draw.
|
|
|
|
|
|
|
|
|
|
Outputs:
|
|
|
|
@ -594,11 +594,11 @@ class UniformCandidateSampler(PrimitiveWithInfer):
|
|
|
|
|
|
|
|
|
|
Outputs:
|
|
|
|
|
- **sampled_candidates** (Tensor) - The sampled_candidates is independent of the true classes.
|
|
|
|
|
Shape: (num_sampled, ).
|
|
|
|
|
Shape: (num_sampled, ).
|
|
|
|
|
- **true_expected_count** (Tensor) - The expected counts under the sampling distribution of each
|
|
|
|
|
of true_classes. Shape: (batch_size, num_true).
|
|
|
|
|
of true_classes. Shape: (batch_size, num_true).
|
|
|
|
|
- **sampled_expected_count** (Tensor) - The expected counts under the sampling distribution of
|
|
|
|
|
each of sampled_candidates. Shape: (num_sampled, ).
|
|
|
|
|
each of sampled_candidates. Shape: (num_sampled, ).
|
|
|
|
|
|
|
|
|
|
Supported Platforms:
|
|
|
|
|
``GPU``
|
|
|
|
|