Function torch::nn::init::kaiming_uniform_#

Function Documentation#

Tensor torch::nn::init::kaiming_uniform_(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 uniform distribution. Also known as He initialization. No gradient will be recorded for this operation.