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