Beta Distribution in PyTorch for a, b>1?



PyTorch支持Beta发行版,但是,当alpha或beta大于1时,它不起作用:

m = Beta(torch.tensor([2]), torch.tensor([2]))
m.sample()

它按预期工作,使用FloatTensortorch==0.4.0

import torch
from torch.distributions import Beta
m = Beta(torch.FloatTensor([2]), torch.FloatTensor([2]))
m.sample()

最新更新