def randn(*args):
  """Returns samples from a normal distribution.

  Uses `tf.random_normal`.

  Args:
    *args: The shape of the output array.

  Returns:
    An ndarray with shape `args` and dtype `float64`.
  """
  # TODO(wangpeng): Use new stateful RNG
  if utils.isscalar(args):
    args = (args,)
  return utils.tensor_to_ndarray(
      tf.random.normal(args, dtype=DEFAULT_RANDN_DTYPE))
Returns samples from a normal distribution.

  Uses `tf.random_normal`.

  Args:
    *args: The shape of the output array.

  Returns:
    An ndarray with shape `args` and dtype `float64`.
  
trax/tf_numpy/numpy_impl/random.py