Функция debugging.assert_equal

Функция debugging.assert_equal проверяет, что два переданных тензора равны друг другу. Первым параметром передается первый тензор, вторым - второй тензор. Если значения тензоров не совпадают, функция выбрасывает исключение InvalidArgumentError. Дополнительно можно передать сообщение об ошибке через параметр message и имя через параметр name. Функция часто используется внутри графа вычислений для отладки и проверки корректности данных.

Синтаксис

tf.debugging.assert_equal(x, y, [message], [name])

Пример

Давайте проверим равенство двух одинаковых тензоров:

import tensorflow as tf

x = tf.constant([1, 2, 3, 4, 5])
y = tf.constant([1, 2, 3, 4, 5])

res = tf.debugging.assert_equal(x, y)
print(res)

Результат выполнения кода:

None

Пример

Давайте проверим равенство двух разных тензоров с пользовательским сообщением об ошибке:

import tensorflow as tf

x = tf.constant([1, 2, 3, 4, 5])
y = tf.constant([1, 2, 3, 4, 6])

res = tf.debugging.assert_equal(x, y, message="tensors are not equal")
print(res)

Результат выполнения кода:

tensorflow.python.framework.errors_impl.InvalidArgumentError: tensors are not equal

Пример

Давайте проверим равенство двух одинаковых двумерных тензоров:

<+python+> import tensorflow as tf x = tf.constant([[1, 2, 3], [4, 5, 6]]) y = tf.constant([[1, 2, 3], [4, 5, 6]]) res = tf.debugging.assert_equal(x, y) print(res) <-python+>

Результат выполнения кода:

None

Смотрите также

функцию debugging.assert_near,
которая проверяет близость двух тензоров
функцию debugging.assert_greater,
которая проверяет, что один тензор больше другого
функцию debugging.assert_less,
которая проверяет, что один тензор меньше другого
функцию debugging.assert_shapes,
которая проверяет формы тензоров