message: Update random_binomial to take number of trials.
  Errors out if number of trials is larger than 1.
diff: @@ -5093,6 +5093,10 @@ def random_uniform(shape, minval=0.0, maxval=1.0, dtype=None, seed=None):
  def random_binomial(shape, p=0.0, dtype=None, seed=None):
  """Returns a tensor with random binomial distribution of values.
  + The binomial distribution with parameters `n` and `p` is the probability
  + distribution of the number of successful Bernoulli process. Only supports
  + `n` = 1 for now.
  +
  Arguments:
  shape: A tuple of integers, the shape of tensor to create.
  p: A float, `0. <= p <= 1`, probability of binomial distribution.
  