2022-08-17 09:53:55 +08:00
|
|
|
|
mindspore.ops.RandomCategorical
|
|
|
|
|
|
===============================
|
|
|
|
|
|
|
|
|
|
|
|
.. py:class:: mindspore.ops.RandomCategorical(dtype=mstype.int64)
|
|
|
|
|
|
|
|
|
|
|
|
从分类分布中抽取样本。
|
|
|
|
|
|
|
2022-08-18 09:54:14 +08:00
|
|
|
|
参数：
|
|
|
|
|
|
- **dtype** (mindspore.dtype) - 输出的类型。它的值必须是 mindspore.int16、mindspore.int32 和 mindspore.int64 之一。默认值：mindspore.int64。
|
2022-08-17 09:53:55 +08:00
|
|
|
|
|
2022-08-18 09:54:14 +08:00
|
|
|
|
输入：
|
2022-09-05 17:10:22 +08:00
|
|
|
|
- **logits** (Tensor) - 输入Tensor。Shape为 :math:`(batch\_size, num\_classes)` 的二维Tensor。
|
2022-08-18 09:54:14 +08:00
|
|
|
|
- **num_sample** (int) - 要抽取的样本数。只允许使用常量值。
|
|
|
|
|
|
- **seed** (int) - 随机种子。只允许使用常量值。默认值：0。
|
2022-08-17 09:53:55 +08:00
|
|
|
|
|
2022-08-18 09:54:14 +08:00
|
|
|
|
输出：
|
2022-08-17 09:53:55 +08:00
|
|
|
|
- **output** (Tensor) - Shape为[batch_size, num_samples]的输出Tensor。
|
|
|
|
|
|
|
2022-08-18 09:54:14 +08:00
|
|
|
|
异常：
|
2022-08-17 09:53:55 +08:00
|
|
|
|
- **TypeError** - 如果 `dtype` 不是以下之一：mindspore.int16、mindspore.int32、mindspore.int64。
|
|
|
|
|
|
- **TypeError** - 如果 `logits` 不是Tensor。
|
|
|
|
|
|
- **TypeError** - 如果 `num_sample` 或者 `seed` 不是 int。
|
