Python PyTorch HalfNormal用法及代碼示例

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

用法:

class torch.distributions.half_normal.HalfNormal(scale, validate_args=None)

參數：

scale(float或者Tensor) -完全正態分布的尺度

基礎：torch.distributions.transformed_distribution.TransformedDistribution

創建由 scale 參數化的半正態分布，其中：

X ~ Normal(0, scale)
Y = |X| ~ HalfNormal(scale)

例子：

>>> m = HalfNormal(torch.tensor([1.0]))
>>> m.sample()  # half-normal distributed with scale=1
tensor([ 0.1046])