initializer

Documentation

Index

• func CalculateFanInAndFanOut(input torch.Tensor) (int64, int64)
• func KaimingUniform(input *torch.Tensor, a float64, fanMode string, nonLinearity string)
• func ManualSeed(seed int64)
• func Normal(a *torch.Tensor, mean, std float64)
• func Ones(a *torch.Tensor)
• func Uniform(a *torch.Tensor, low, high float64)
• func Zeros(a *torch.Tensor)

Functions

func CalculateFanInAndFanOut

func CalculateFanInAndFanOut(input torch.Tensor) (int64, int64)

CalculateFanInAndFanOut torch.nn.init._calculate_fan_in_and_fan_out

func KaimingUniform

func KaimingUniform(input *torch.Tensor, a float64, fanMode string,
	nonLinearity string)

KaimingUniform initialization, torch.nn.init.kaiming_uniform_

func ManualSeed

func ManualSeed(seed int64)

ManualSeed set the random seed

func Normal

func Normal(a *torch.Tensor, mean, std float64)

Normal initialization, torch.nn.init.normal_

func Ones

func Ones(a *torch.Tensor)

Ones initialization, torch.nn.init.ones_

func Uniform

func Uniform(a *torch.Tensor, low, high float64)

Uniform initialization, torch.nn.init.uniform_

func Zeros

func Zeros(a *torch.Tensor)

Zeros initialization, torch.nn.init.zeros_