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=-1) – The value to \(\alpha\).
  • b (number, optional, default=1) – The value to \(\beta\).
Returns:

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