tf.random.categorical()用法解析

从一个分类分布中抽取样本(索引对应的概率服从多项分布)，输出分类的index

tf.random.categorical(
    logits,#形状为 [batch_size, num_classes]的张量. 每个切片 [i, :] 代表对于所有类的未正规化的log概率。最后softmax的概率分布；也可以是整数，会自动变换成概率分布
    num_samples,#抽取的样本个数
    dtype=None,
    seed=None,
    name=None
)