Python PyTorch Uniform用法及代碼示例

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

用法:

class torch.distributions.uniform.Uniform(low, high, validate_args=None)

參數：

• low(float或者Tensor) -較低的範圍(包括)。

• high(float或者Tensor) -上限(不包括)。

基礎：torch.distributions.distribution.Distribution

從半開區間 [low, high) 生成均勻分布的隨機樣本。

例子：

>>> m = Uniform(torch.tensor([0.0]), torch.tensor([5.0]))
>>> m.sample()  # uniformly distributed in the range [0.0, 5.0)
tensor([ 2.3418])