Python PyTorch Gumbel用法及代碼示例

本文簡要介紹python語言中 torch.distributions.gumbel.Gumbel 的用法。

用法:

class torch.distributions.gumbel.Gumbel(loc, scale, validate_args=None)

參數：

• loc(float或者Tensor) -分布的位置參數

• scale(float或者Tensor) -分布的尺度參數

基礎：torch.distributions.transformed_distribution.TransformedDistribution

來自 Gumbel 分布的樣本。

例子：

>>> m = Gumbel(torch.tensor([1.0]), torch.tensor([2.0]))
>>> m.sample()  # sample from Gumbel distribution with loc=1, scale=2
tensor([ 1.0124])