Function torch::nn::init::kaiming_normal_

Function Documentation

Tensor torch::nn::init::kaiming_normal_(Tensor tensor, double a = 0, FanModeType mode = torch::kFanIn, NonlinearityType nonlinearity = torch::kLeakyReLU)

Fills the input Tensor

with values according to the method described in “Delving deep into rectifiers: Surpassing human-level

performance on ImageNet classification” - He, K.

et al. (2015), using a normal distribution. Also known as He initialization. No gradient will be recorded for this operation.


Access comprehensive developer documentation for PyTorch

View Docs


Get in-depth tutorials for beginners and advanced developers

View Tutorials


Find development resources and get your questions answered

View Resources