PyTorch 中 a、b>1 的 Beta 分布？

Question

PyTorch 支持 Beta distributions 但是，当 alpha 或 beta 大于 1 时，它不起作用：

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

Answer 1

它按预期使用 FloatTensor 和 torch==0.4.0:

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