PyTorch支持Beta发行版,但是,当alpha或beta大于1时,它不起作用:
m = Beta(torch.tensor([2]), torch.tensor([2]))
m.sample()
它按预期工作,使用FloatTensor
和torch==0.4.0
:
import torch
from torch.distributions import Beta
m = Beta(torch.FloatTensor([2]), torch.FloatTensor([2]))
m.sample()