Google I / O kehrt vom 18. bis 20. Mai zurück! Reservieren Sie Platz und erstellen Sie Ihren Zeitplan Registrieren Sie sich jetzt

tf.debugging.assert_rank

TensorFlow 2 version View source on GitHub

Assert x has rank equal to rank.

Example of adding a dependency to an operation:

with tf.control_dependencies([tf.compat.v1.assert_rank(x, 2)]):
  output = tf.reduce_sum(x)

x Numeric Tensor.
rank Scalar integer Tensor.
data The tensors to print out if the condition is False. Defaults to error message and the shape 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".

Op raising InvalidArgumentError unless x has specified rank. If static checks determine x has correct rank, a no_op is returned.

ValueError If static checks determine x has wrong rank.