方法有两种:
-
torch.multinomial函数
使用方法和正太分布没太大区别 -
torch.distributions.Categorical().sample函数
p = torch.distributions.Categorical(torch.tensor([0.25, 0.25, 0.25, 0.25]))
print(p.entropy())
方法有两种:
torch.multinomial函数
使用方法和正太分布没太大区别
torch.distributions.Categorical().sample函数
p = torch.distributions.Categorical(torch.tensor([0.25, 0.25, 0.25, 0.25]))
print(p.entropy())