Функция debugging.assert_integer

Функция debugging.assert_integer проверяет, что переданный тензор имеет целочисленный тип данных. Первым параметром функция принимает тензор для проверки. Вторым параметром можно передать сообщение об ошибке, которое будет выведено при невыполнении условия.

Синтаксис

tf.debugging.assert_integer(x, [message])

Пример

Давайте создадим тензор с целочисленным типом и проверим его с помощью функции:

import tensorflow as tf

t = tf.constant([1, 2, 3, 4, 5])
res = tf.debugging.assert_integer(t)
print(res)

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

None

Пример

Давайте создадим тензор с типом float32 и попробуем проверить его. Функция должна выбросить исключение:

import tensorflow as tf

t = tf.constant([1.0, 2.0, 3.0])
res = tf.debugging.assert_integer(t)
print(res)

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

InvalidArgumentError: Expected 'x' to be an integer tensor, got float32 instead.

Пример

Давайте передадим собственное сообщение об ошибке через второй параметр:

import tensorflow as tf

t = tf.constant([1.0, 2.0, 3.0])
res = tf.debugging.assert_integer(t, message="tensor must be integer")
print(res)

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

InvalidArgumentError: tensor must be integer

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

функцию debugging.assert_equal,
которая проверяет равенство двух тензоров
функцию debugging.assert_type,
которая проверяет тип данных тензора
функцию debugging.assert_rank,
которая проверяет ранг тензора
функцию debugging.assert_shapes,
которая проверяет форму тензоров