tensorplay.nn.init.trunc_normal_

tensorplay.nn.init.trunc_normal_(tensor, mean=0.0, std=1.0, a=-2.0, b=2.0, generator=None)[source]

Fills the input Tensor with values drawn from a truncated normal distribution.

Method is based on the rejection-sampling scheme in https://people.sc.fsu.edu/~jburkardt/presentations/truncated_normal.pdf —