如何在pytorch中制作截断正态分布？

【问题标题】：How to make a Truncated normal distribution in pytorch?如何在pytorch中制作截断正态分布？
【发布时间】：2020-02-14 20:37:54
【问题描述】：

我想在 PyTorch 中创建一个截断正态分布（即具有范围的高斯分布）。
我希望能够更改均值、标准差和范围。
有 PyTorch 方法吗？

【问题讨论】：

这似乎有帮助，你基本上可以复制等效的scipy版本：discuss.pytorch.org/t/implementing-truncated-normal-initializer/…

标签： pytorch normal-distribution

【解决方案1】：

使用torch.nn.init.trunc_normal_。

描述如Here:

用截断的值填充输入张量正态分布。这些值有效地从正态分布：数学：\mathcal{N}(\text{mean}, \text{std}^2) 重绘 :math:[a, b] 之外的值，直到它们在范围内界限。用于生成随机值的方法有效最好的时候:math:a \leq \text{mean} \leq b.

【讨论】：