code: def truncated_normal(
      shape: Tuple[int, ...],
      *,
      stddev: float = 1,
      lower: float = -2,
      upper: float = 2,
      generator: Generator = DEFAULT_GENERATOR,
  ):
      """Returns a ``JaxArray`` of shape ``shape`` with random numbers from a normal distribution
      with mean 0 and standard deviation ``stddev`` truncated by (``lower``, ``upper``).
  
      NOTE: if random numbers are generated inside a jitted, parallelized or vectorized function
      then generator variables (including DEFAULT_GENERATOR) have to be added to the
      variable collection."""
      return (
          jr.truncated_normal(
              generator(), shape=shape, lower=lower, upper=upper
          )
          * stddev
      )
  
file_path: objax/objax/random/random.py
imports: from typing import Optional, Tuple
  import jax.random
  from objax.module import Module
  from objax.util import class_name
  from objax.variable import RandomState, VarCollection
label_desc: Use image node_id 3 for calling a global function with example usage: truncated_normal(shape) and returns: unknown