Function torch::nn::init::kaiming_normal_¶
Defined in File init.h
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.