normal#

torchdyno.models.initializers.normal(shape, mean=0, std=1, dtype=torch.float32)[source]#

Normal random tensor. Can either be rescaled according to spectral radius rho, spectral norm sigma, or scale.

Parameters:

  • shape (torch.Size) – shape of the tensor.

  • mean (float, optional) – mean value for normal initialization. Defaults to 0.

  • std (float, optional) – standard deviation for normal initialization. Defaults to 1.

  • dtype (dtype)

Returns:

initialized tensor.

Return type:

torch.Tensor