torch.nn.init — PyTorch 1.10.1 documentation
pytorch.org › docs › stabletorch.nn.init.dirac_(tensor, groups=1) [source] Fills the {3, 4, 5}-dimensional input Tensor with the Dirac delta function. Preserves the identity of the inputs in Convolutional layers, where as many input channels are preserved as possible. In case of groups>1, each group of channels preserves identity. Parameters.