在输入x中根据每个对象分布的概率,抽取numSamples个样本,并将这些样本的索引存储在输出y中。
x:输入Tensor,shape=[batch_size, class_size],数据类型支持float16、float。class_size 指所有可能结果的数量,每个值表示该batch中每个相应结果的非归一化对数概率。
y:输出Tensor,shape=[batch_size, sample_size],数据类型支持int32、int64。sample_size指采样的次数,每个值表示该batch中相应样本的结果。
如果调用该算子超时,需要使用ret = acl.rt.set_op_execute_time_out(timeout)接口避免超时。
无。
Opset v8/v9/v10/v11/v12/v13/v14/v15/v16/v17/v18。