with tf.control_dependencies([tf.compat.v1.assert_rank(x, 2)]):
output = tf.reduce_sum(x)
Аргументы
x
Числовое значение Tensor.
rank
Скалярное целое число Tensor.
data
Тензоры для вывода, если условие ложно. По умолчанию сообщение об ошибке и форма x.
summarize
Количество элементов каждого тензора для вывода.
message
Строка для добавления к стандартному сообщению.
name
Имя данной операции (необязательно). По умолчанию "assert_rank".
Возвращаемое значение
Операция, которая вызывает InvalidArgumentError , если x имеет указанный ранг. Если статические проверки показывают, что у x правильный ранг, возвращается no_op.
Исключения
ValueError
Если статические проверки показывают, что у x неправильный ранг.