这个屌丝很懒，什么也没留下！

Pytorch probability distributions_from torch.distributions.one_hot_categorical impor

作者：我家小花儿 | 2024-04-01 23:17:11

from torch.distributions.one_hot_categorical import onehotcategorical

torch.distributions.one_hot_categorical.OneHotCategorical(probs=None, logits=None, validate_args=None)

根据给定的概率probs, 创建一个 one-hot 的类别分布.

m = OneHotCategorical(torch.tensor([ 0.1, 0.0, 0.9, 0.0 ]))
m.sample()  # equal probability of 0, 1, 2, 3
#tensor([0., 0., 1., 0.])
1
2
3

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/我家小花儿/article/detail/350378