parameters: (tensor, expected_rank, name=None)
docstring: Raises an exception if the tensor rank is not of the expected rank.
  
    Args:
      tensor: A tf.Tensor to check the rank of.
      expected_rank: Python integer or list of integers, expected rank.
      name: Optional name of the tensor for the error message.
  
    Raises:
      ValueError: If the expected shape doesn't match the actual shape.
docstring_summary: Raises an exception if the tensor rank is not of the expected rank.
docstring_tokens: Raises an exception if the tensor rank is not of the expected rank .
function: def assert_rank(tensor, expected_rank, name=None):
    """Raises an exception if the tensor rank is not of the expected rank.
  
    Args:
      tensor: A tf.Tensor to check the rank of.
      expected_rank: Python integer or list of integers, expected rank.
      name: Optional name of the tensor for the error message.
  
    Raises:
      ValueError: If the expected shape doesn't match the actual shape.
    """
    if name is None:
      name = tensor.name
  
    expected_rank_dict = {}
    if isinstance(expected_rank, six.integer_types):
      expected_rank_dict[expected_rank] = True
    else:
      for x in expected_rank:
        expected_rank_dict[x] = True
  
    actual_rank = tensor.shape.ndims
    if actual_rank not in expected_rank_dict:
      scope_name = tf.get_variable_scope().name
      raise ValueError(
          "For the tensor `%s` in scope `%s`, the actual rank "
          "`%d` (shape = %s) is not equal to the expected rank `%s`" %
          (name, scope_name, actual_rank, str(tensor.shape), str(expected_rank)))
function_tokens: def assert_rank ( tensor , expected_rank , name = None ) : if name is None : name = tensor . name expected_rank_dict = { } if isinstance ( expected_rank , six . integer_types ) : expected_rank_dict [ expected_rank ] = True else : for x in expected_rank : expected_rank_dict [ x ] = True actual_rank = tensor . shape . ndims if actual_rank not in expected_rank_dict : scope_name = tf . get_variable_scope ( ) . name raise ValueError ( "For the tensor `%s` in scope `%s`, the actual rank " "`%d` (shape = %s) is not equal to the expected rank `%s`" % ( name , scope_name , actual_rank , str ( tensor . shape ) , str ( expected_rank ) ) )