W3cubDocs

/TensorFlow 2.4

tf.compat.v1.assert_rank_in

Assert x has rank in ranks.

Example of adding a dependency to an operation:

with tf.control_dependencies([tf.compat.v1.assert_rank_in(x, (2, 4))]):
  output = tf.reduce_sum(x)
Args
x Numeric Tensor.
ranks Iterable of scalar Tensor objects.
data The tensors to print out if the condition is False. Defaults to error message and first few entries of x.
summarize Print this many entries of each tensor.
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.
Raises
ValueError If static checks determine x has mismatched rank.

© 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/compat/v1/assert_rank_in