uniform_¶
dragon.vm.torch.nn.init.
uniform_
(
tensor,
a=0,
b=1
)[source]¶Fill tensor from an uniform distribution.
\[\text{tensor} \sim \mathcal{U}(\alpha, \beta) \]- Parameters:
- tensor (dragon.vm.torch.Tensor) – The input tensor.
- a (number, optional, default=0) – The value to \(\alpha\).
- b (number, optional, default=1) – The value to \(\beta\).
- Returns:
dragon.vm.torch.Tensor – The output tensor.