Функция assert_equal

Функция assert_equal применяется для проверки равенства двух тензоров. Первым параметром передаётся первый тензор, вторым - второй тензор. Третьим необязательным параметром можно передать сообщение об ошибке, которое будет выведено при несовпадении значений.

Синтаксис

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

Пример

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

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)

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

InvalidArgumentError: tensors are not equal

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

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