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.