tf.debugging.assert_equal

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

tf.debugging.assert_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_equal(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