Модуль: tf.test

Тестирование.

Классы

class Benchmark: Абстрактный класс, предоставляющий вспомогательные средства для бенчмарков TensorFlow.

class TestCase: Базовый класс для тестов, требующих проверки TensorFlow.

Функции

assert_equal_graph_def(...): Утверждает, что две GraphDef (в основном) одинаковые.

benchmark_config(...): Возвращает tf.compat.v1.ConfigProto для отключения оптимизатора зависимостей.

compute_gradient(...): Вычисляет теоретический и числовой якобиан f.

create_local_cluster(...): Создаёт и запускает локальные серверы и возвращает связанные Server объекты.

disable_with_predicate(...): Отключает тест, если pred истинно.

gpu_device_name(...): Возвращает имя устройства GPU, если оно доступно, или пустую строку.

is_built_with_cuda(...): Возвращает, было ли TensorFlow скомпилировано с поддержкой CUDA (GPU).

is_built_with_gpu_support(...): Возвращает, было ли TensorFlow скомпилировано с поддержкой GPU (CUDA или ROCm).

is_built_with_rocm(...): Возвращает, было ли TensorFlow скомпилировано с поддержкой ROCm (GPU).

is_built_with_xla(...): Возвращает, было ли TensorFlow скомпилировано с поддержкой XLA.

is_gpu_available(...): Возвращает, может ли TensorFlow получить доступ к GPU. (устарело)

main(...): Запускает все unit-тесты.

with_eager_op_as_function(...): Добавляет методы, которые вызывают оригинальные методы с включённым eager_op_as_function.