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 output tensor.