You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

mindspore.ops.UniformCandidateSampler.rst 1.9 kB

4 years ago
1234567891011121314151617181920212223242526272829303132333435
  1. mindspore.ops.UniformCandidateSampler
  2. ======================================
  3. .. py:class:: mindspore.ops.UniformCandidateSampler(num_true, num_sampled, unique, range_max, seed=0, remove_accidental_hits=False)
  4. 使用均匀分布对一组类别进行采样。
  5. 此函数使用均匀分布从[0, range_max-1]中采样一组类(sampled_candidates)。如果 `unique` 为True,则候选采样没有重复;如果 `unique` 为False,则有重复。
  6. **参数:**
  7. - **num_true** (int) - 每个训练样本的目标类数。
  8. - **num_sampled** (int) - 随机采样的类数。sampled_candidates的shape将为 `num_sampled` 。如果 `unique` 为True,则 `num_sampled` 必须小于或等于 `range_max` 。
  9. - **unique** (bool) - 表示一个batch中的所有采样类是否唯一。
  10. - **range_max** (int) - 可能的类数,该值必须是非负的。
  11. - **seed** (int) - 随机种子,该值必须是非负的。如果seed的值为0,则seed的值将被随机生成的值替换。默认值:0。
  12. - **remove_accidental_hits** (bool) - 表示是否移除accidental hit。默认值:False。
  13. **输入:**
  14. - **true_classes** (Tensor) - 输入Tensor,目标类,其shape为(batch_size, num_true)。
  15. **输出:**
  16. - **sampled_candidates** (Tensor) - 候选采样与目标类之间不存在联系,其shape为(num_sampled, )。
  17. - **true_expected_count** (Tensor) - 在每组目标类的采样分布下的预期计数。Shape为(batch_size, num_true)。
  18. - **sampled_expected_count** (Tensor) - 每个候选采样分布下的预期计数。Shape为(num_sampled, )。
  19. **异常:**
  20. - **TypeError** - `num_true` 和 `num_sampled` 都不是int。
  21. - **TypeError** - `uique` 和 `remo_acidental_hits` 都不是bool。
  22. - **TypeError** - `range_max` 和 `seed` 都不是int。
  23. - **TypeError** - `true_classes` 不是Tensor。