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