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.