tf.random.categorical()用法解析
从一个分类分布中抽取样本(索引对应的概率服从多项分布)，输出分类的index
tf.random.categorical(
logits,#形状为 [batch_size, num_classes]的张量. 每个切片 [i, :] 代表对于所有类的未正规化的log概率。最后softmax的概率分布；也可以是整数，会自动变换成概率分布
num_samples,#抽取的样本个数
dtype=None,
seed=None,
name=None
)
image.png
本文参与 腾讯云自媒体分享计划，分享自作者个人站点/博客。
如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读