Распределение бета в PyTorch для a, b> 1?

Question

PyTorch поддерживает бета-дистрибутивы однако, когда альфа или бета больше 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()