Shortcuts

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.

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources