Truncated normal initialization
Fills the input Tensor with values drawn from a truncated normal distribution.
nn_init_trunc_normal_(tensor, mean = 0, std = 1, a = -2, b = 2)
tensor |
an n-dimensional Tensor |
mean |
the mean of the normal distribution |
std |
the standard deviation of the normal distribution |
a |
the minimum cutoff value |
b |
the maximum cutoff value |
if (torch_is_installed()) { w <- torch_empty(3, 5) nn_init_trunc_normal_(w) }
Please choose more modern alternatives, such as Google Chrome or Mozilla Firefox.