torch.distributions.multinomial.Multinomial(total_count=1probs=Nonelogits=Nonevalidate_args=None)

看书看到这个函数不是很懂,搜索发现连 torch API 都没有很细致地讲,我最后的理解如下:

比方说以下面为例,这里有几个前提:

1、probs.sum() 为1

2、probs有几个元素,我们就只能往几个位置掷骰子,这是限定的大前提

3、total_count就是投的次数。根据大数定律,投的次数越多,sample结果的分布概率越接近prob。

Logo

开放原子开发者工作坊旨在鼓励更多人参与开源活动,与志同道合的开发者们相互交流开发经验、分享开发心得、获取前沿技术趋势。工作坊有多种形式的开发者活动,如meetup、训练营等,主打技术交流,干货满满,真诚地邀请各位开发者共同参与!

更多推荐