initial commit, fixed indentation of inputs/outputs

pull/9403/head
Peilin Wang 5 years ago
parent b15504f6a5
commit acfe9c9aab

@ -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