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