truncated_normal¶

dragon.random.truncated_normal(
  shape,
  mean=0,
  std=1,
  dtype='float32',
  **kwargs
)[source]¶

Return a tensor initialized from the truncated normal distribution.

\[\text{out} \sim \mathcal{TN}(\mu, \sigma^{2}, \mu - 2\sigma, \mu + 2\sigma) \]

Parameters:

• shape (Sequence[Union[int, dragon.Tensor]]) – The tensor shape.
• mean (number, optional, default=0) – The value to \(\mu\).
• std (number, optional, default=1) – The value to \(\sigma\).
• dtype (str, optional, default='float32') – The optional data type.

Returns:

dragon.Tensor – The output tensor.