tf.random.categorical()用法解析
2020-11-21 本文已影响0人
飘涯
从一个分类分布中抽取样本(索引对应的概率服从多项分布)，输出分类的index
tf.random.categorical(
logits,#形状为 [batch_size, num_classes]的张量. 每个切片 [i, :] 代表对于所有类的未正规化的log概率。最后softmax的概率分布；也可以是整数，会自动变换成概率分布
num_samples,#抽取的样本个数
dtype=None,
seed=None,
name=None
)
image.png
