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 применяется во время выполнения. Рекомендуется использовать его вместе с 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