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