(n, new_shape, *arys)
Reshape arrays to be at least `n`-dimensional.

  Args:
    n: The minimal rank.
    new_shape: a function that takes `n` and the old shape and returns the
      desired new shape.
    *arys: ndarray(s) to be reshaped.

  Returns:
    The reshaped array(s).
Reshape arrays to be at least `n`-dimensional.
Reshape arrays to be at least n - dimensional .
def _atleast_nd(n, new_shape, *arys):
  """Reshape arrays to be at least `n`-dimensional.

  Args:
    n: The minimal rank.
    new_shape: a function that takes `n` and the old shape and returns the
      desired new shape.
    *arys: ndarray(s) to be reshaped.

  Returns:
    The reshaped array(s).
  """

  def f(x):
    # pylint: disable=g-long-lambda
    x = asarray(x)
    return asarray(
        utils.cond(
            utils.greater(n, tf.rank(x)),
            lambda: reshape(x, new_shape(n, tf.shape(x.data))).data,
            lambda: x.data))

  arys = list(map(f, arys))
  if len(arys) == 1:
    return arys[0]
  else:
    return arys
def _atleast_nd ( n , new_shape , * arys ) : def f ( x ) : # pylint: disable=g-long-lambda x = asarray ( x ) return asarray ( utils . cond ( utils . greater ( n , tf . rank ( x ) ) , lambda : reshape ( x , new_shape ( n , tf . shape ( x . data ) ) ) . data , lambda : x . data ) ) arys = list ( map ( f , arys ) ) if len ( arys ) == 1 : return arys [ 0 ] else : return arys