tf.debugging.assert_rank_in

Assert that x has a rank in ranks.

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

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/debugging/assert_rank_in