Defined in File init.h
kaiming_uniform_(Tensor tensor, double a = 0, FanModeType mode = torch::kFanIn, NonlinearityType nonlinearity = torch::kLeakyReLU)¶
Fills the input
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.