tf.debugging.assert_less

Проверка условия x < y выполняется поэлементно.

tf.debugging.assert_less(
    x, y, message=None, summarize=None, name=None
)

Этот оператор проверяет, что x[i] < y[i] выполняется для каждой пары (возможно, с вещанием) элементов из x и y. Если оба x и y пусты, это выполняется тривиально.

Если x < y не выполняется, message, а также первые summarize элементы из x и y выводятся на экран, и происходит поднятие исключения InvalidArgumentError.

При использовании внутри tf.function этот API срабатывает во время выполнения. Рекомендуется использовать этот API с tf.control_dependencies для обеспечения правильного порядка выполнения.

В следующем примере, без tf.control_dependencies, ошибки могут вообще не возникнуть. Подробнее см. tf.control_dependencies.

def check_size(x):
  with tf.control_dependencies([
      tf.debugging.assert_less(tf.size(x), 3,
                      message='Bad tensor size')]):
    return x

check_size(tf.ones([2, 3], tf.float32))
Traceback (most recent call last):

InvalidArgumentError: ...

возвращает None