当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

Python tf.random.categorical用法及代码示例


从分类分布中抽取样本。

用法

tf.random.categorical(
    logits, num_samples, dtype=None, seed=None, name=None
)

参数

  • logits 形状为 [batch_size, num_classes] 的二维张量。每个切片 [i,:] 代表所有类的非规范化 log-probabilities。
  • num_samples 0-D。为每个行切片绘制的独立样本数。
  • dtype 输出的整数类型：int32 或 int64。默认为 int64 。
  • seed 一个 Python 整数。用于为分发创建随机种子。有关行为，请参见tf.random.set_seed。
  • name 操作的可选名称。

返回

  • 形状 [batch_size, num_samples] 的绘制样本。

例子：

# samples has shape [1, 5], where each value is either 0 or 1 with equal
# probability.
samples = tf.random.categorical(tf.math.log([[0.5, 0.5]]), 5)

相关用法


注：本文由纯净天空筛选整理自tensorflow.org大神的英文原创作品 tf.random.categorical。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。