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