Truncated normal initialization (approximation)
trunc_normal_(x, mean = 0, std = 1)
x
tensor
mean
std
standard deviation