MindSpore官方RandomChoiceWithMask算子用例报错

mindspore.ops 算子 RandomChoiceWithMask

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)

【日志信息】（可选，上传日志内容或者附件）

Traceback (most recent call last):
  File "E:/Mindspore1.1/mindspore/model_zoo/official/cv/faster_rcnn/test.py", line 21, in
    output_y, output_mask = rnd_choice_mask(input_x)
  File "F:\py375\lib\site-packages\mindspore\ops\primitive.py", line 247, in __call__
    return _run_op(self, self.name, args)
  File "F:\py375\lib\site-packages\mindspore\common\api.py", line 77, in wrapper
    results = fn(*arg, **kwargs)
  File "F:\py375\lib\site-packages\mindspore\ops\primitive.py", line 682, in _run_op
    output = real_run_op(obj, op_name, args)
RuntimeError: mindspore\ccsrc\runtime\device\cpu\cpu_kernel_runtime.cc:469 Run] mindspore/core/utils/convert_utils_base.h:28 SizeToInt] The size_t value(3499211612) exceeds the maximum value of int.

# 
Trace:

# 
WARNING: Logging before InitGoogleLogging() is written to STDERR
[EXCEPTION] PRE_ACT(60896,3,?):2021-10-14 16:18:30 [mindspore/core/utils/convert_utils_base.h:28] SizeToInt] The size_t value(3499211612) exceeds the maximum value of int.
[EXCEPTION] DEVICE(60896,3,?):2021-10-14 16:18:30 [mindspore\ccsrc\runtime\device\cpu\cpu_kernel_runtime.cc:469] Run] mindspore/core/utils/convert_utils_base.h:28 SizeToInt] The size_t value(3499211612) exceeds the maximum value of int.

# 
Trace:
[WARNING] PIPELINE(60896,1,?):2021-10-14 16:18:30 [mindspore\ccsrc\pipeline\jit\init.cc:310] operator()] Start releasing dataset handles...
[WARNING] PIPELINE(60896,1,?):2021-10-14 16:18:30 [mindspore\ccsrc\pipeline\jit\init.cc:313] operator()] End release dataset handles.

 

https://gitee.com/mindspore/mindspore/pulls/24880
https://gitee.com/mindspore/mindspore/pulls/24881

更正：该问题的修复pr链接如上，合入时间为10月15日。