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

Question

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

Answer 1

使用 torch.nn.init.trunc_normal_ .
给出的描述 Here :

Fills the input Tensor with values drawn from a truncatednormal distribution. The values are effectively drawn from thenormal distribution :math:\mathcal{N}(\text{mean}, \text{std}^2)with values outside :math:[a, b] redrawn until they are withinthe bounds. The method used for generating the random values worksbest when :math:a \leq \text{mean} \leq b.