pytorch中随机生成cat(类别分布,multinomial分布),来训练gan网络的方法

方法有两种:

  1. torch.multinomial函数
    使用方法和正太分布没太大区别

  2. torch.distributions.Categorical().sample函数
    p = torch.distributions.Categorical(torch.tensor([0.25, 0.25, 0.25, 0.25]))
    print(p.entropy())

你可能感兴趣的:(pytorch)