Модуль: tf.compat.v2.test

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

См. руководство по тестированию.

Примечание: tf.compat.v1.test.mock является псевдонимом для python mock или unittest.mock в зависимости от версии python.

Классы

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

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

Функции

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

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

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

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

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_gpu_available(...): Возвращает значение, указывающее, может ли TensorFlow получить доступ к GPU.

main(...): Запускает все модульные тесты.