nn_init_normal_ function

Normal initialization

Normal initialization

Fills the input Tensor with values drawn from the normal distribution

nn_init_normal_(tensor, mean = 0, std = 1)

Arguments

  • tensor: an n-dimensional Tensor
  • mean: the mean of the normal distribution
  • std: the standard deviation of the normal distribution

Examples

if (torch_is_installed()) { w <- torch_empty(3, 5) nn_init_normal_(w) }
  • Maintainer: Daniel Falbel
  • License: MIT + file LICENSE
  • Last published: 2025-02-14