Python PyTorch Normal用法及代码示例

本文简要介绍python语言中 torch.distributions.normal.Normal 的用法。

用法:

class torch.distributions.normal.Normal(loc, scale, validate_args=None)

参数：

• loc(float或者Tensor) -分布的平均值(通常称为 mu)

• scale(float或者Tensor) -分布的标准差(通常称为 sigma)

基础：torch.distributions.exp_family.ExponentialFamily

创建由 loc 和 scale 参数化的正态(也称为高斯)分布。

例子：

>>> m = Normal(torch.tensor([0.0]), torch.tensor([1.0]))
>>> m.sample()  # normally distributed with loc=0 and scale=1
tensor([ 0.1046])