View source on GitHub
  
 | 
Assert that x has a rank in ranks.
tf.debugging.assert_rank_in(
    x, ranks, message=None, name=None
)
This Op checks that the rank of x is in ranks.
If x has a different rank, message, as well as the shape of x are
printed, and InvalidArgumentError is raised.
Args | |
|---|---|
x
 | 
Tensor.
 | 
ranks
 | 
Iterable of scalar Tensor objects.
 | 
message
 | 
A string to prefix to the default message. | 
name
 | 
A name for this operation (optional). Defaults to "assert_rank_in". | 
Returns | |
|---|---|
Op raising InvalidArgumentError unless rank of x is in ranks.
If static checks determine x has matching rank, a no_op is returned.
This can be used with tf.control_dependencies inside of tf.functions
to block followup computation until the check has executed.
 | 
Raises | |
|---|---|
InvalidArgumentError
 | 
x does not have rank in ranks, but the rank cannot
be statically determined.
 | 
ValueError
 | 
If static checks determine x has mismatched rank.
 | 
eager compatibility
returns None
    View source on GitHub