我期望从多项分布中抽取的数据也是可重复的。为什么我会得到这种行为，我该如何修复？我试图从一个多项分布绘制，并希望绘制是可复制的。
ulydmbyx1#
将generator传递给torch.multinomial只会告诉pytorch使用哪个generator，它不会重置其状态。试试这个
torch.multinomial
import torch g = torch.Generator().manual_seed(1) p = torch.rand(3, generator=g) initial_state = g.get_state() print(torch.multinomial(p, num_samples=10, replacement=True, generator=g)) g.set_state(initial_state) print(torch.multinomial(p, num_samples=10, replacement=True, generator=g))
1条答案
按热度按时间ulydmbyx1#
将generator传递给
torch.multinomial只会告诉pytorch使用哪个generator，它不会重置其状态。试试这个