tf.debugging.assert_greater_equal

Утверждает, что условие x >= y выполняется поэлементно.

tf.debugging.assert_greater_equal(
    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_greater_equal(tf.size(x), 9,
                      message='Bad tensor size')]):
    return x

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

InvalidArgumentError: ...

возвращает None