dirac_

dragon.vm.torch.nn.init.dirac_(
  tensor,
  groups=1
)[source]

Fill tensor with the dirac delta function.

Parameters:
  • tensor (dragon.vm.torch.Tensor) – The input tensor.
  • groups (number, optional, default=1) – The groups of convolution.
Returns:

dragon.vm.torch.Tensor – The input tensor.