Beta Distribution in PyTorch for a, b>1?

Question

PyTorch supports Beta distributions however, when alpha or beta is greater than 1, it doesn't work:

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()