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.