mindspore.ops.RandomChoiceWithMask
- class mindspore.ops.RandomChoiceWithMask(count=256, seed=0, seed2=0)[source]
Generates a random sample as index tensor with a mask tensor from a given tensor.
Refer to
mindspore.ops.choice_with_mask()
for more details.- Parameters
count (int, optional) – Number of items expected to get and the number must be greater than 0. Default: 256.
seed (int, optional) – Seed is used as entropy source for Random number engines generating pseudo-random numbers. Default: 0.
seed2 (int, optional) – Second seed to avoid collision. Default: 0.
- 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.
- Outputs:
Two tensors, the first one is the index tensor and the other one is the mask tensor.
index (Tensor) - The output shape is 2-D.
mask (Tensor) - The output shape is 1-D.
- Supported Platforms:
Ascend
GPU
CPU
Examples
>>> rnd_choice_mask = ops.RandomChoiceWithMask() >>> input_x = Tensor(np.ones(shape=[240000, 4]).astype(np.bool)) >>> output_y, output_mask = rnd_choice_mask(input_x) >>> result = output_y.shape >>> print(result) (256, 2) >>> result = output_mask.shape >>> print(result) (256,)