Number of Class:	Sample times using gumbel max: Sample!
Click to generate a discrete distribution: Generate!	KL divergence:

    References:
    [1] 漫谈重参数：从正态分布到Gumbel Softmax
    [2] Reparametrization Trick
    [3] Wikipedia: Kullback–Leibler divergence