!9403 Fix input/output docstring indentation for ops in random_ops.py

From: @peilin-wang
Reviewed-by: @robingrosman,@tom__chen
Signed-off-by: @robingrosman
pull/9403/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit 98aac335fa

@ -401,7 +401,7 @@ class RandomChoiceWithMask(PrimitiveWithInfer):
Inputs: Inputs:
- **input_x** (Tensor[bool]) - The input tensor. - **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: Outputs:
Two tensors, the first one is the index tensor and the other one is the mask tensor. 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. seed2 (int): Random seed2, must be non-negative. Default: 0.
Inputs: Inputs:
- **input** (Tensor[float32]) - the input tensor containing the cumsum of probabilities, must be 1 or 2 - **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. - **num_samples** (int32) - number of samples to draw.
Outputs: Outputs:
@ -594,11 +594,11 @@ class UniformCandidateSampler(PrimitiveWithInfer):
Outputs: Outputs:
- **sampled_candidates** (Tensor) - The sampled_candidates is independent of the true classes. - **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 - **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 - **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: Supported Platforms:
``GPU`` ``GPU``

Loading…
Cancel
Save