tf.test.Benchmark
Абстрактный класс, предоставляющий вспомогательные функции для бенчмарков TensorFlow.
tf.test.Benchmark()
Методы
evaluate
evaluate(
tensors
)
Оценивает тензоры и возвращает значения NumPy.
| Аргументы | |
|---|---|
tensors | Тензор или вложенный список/кортеж тензоров. |
| Возвращаемое значение | |
|---|---|
| Числовые значения тензоров. |
is_abstract
@classmethod is_abstract()
report_benchmark
report_benchmark(
iters=None,
cpu_time=None,
wall_time=None,
throughput=None,
extras=None,
name=None,
metrics=None
)
Отчет о бенчмарке.
| Аргументы | |
|---|---|
iters | (необязательно) Количество выполненных итераций |
cpu_time | (необязательно) Медианное или среднее время выполнения CPU в секундах. |
wall_time | (необязательно) Медианное или среднее время выполнения в секундах. |
throughput | (необязательно) Пропускная способность (в МБ/с) |
extras | (необязательно) Словарь, сопоставляющий строковые ключи дополнительной информации о бенчмарке. Значения могут быть либо числами с плавающей точкой, либо значениями, которые могут быть преобразованы в строки. |
name | (необязательно) Переопределение имени BenchmarkEntry с помощью name. В противном случае оно выводится из имени метода верхнего уровня. |
metrics | (необязательно) Список словарей, где каждый словарь содержит следующие ключи: name (обязательно), string, имя метрики value (обязательно), double, минимальное значение метрики min_value (необязательно), double, минимальное допустимое значение метрики max_value (необязательно), double, максимальное допустимое значение метрики |
run_op_benchmark
run_op_benchmark(
sess,
op_or_tensor,
feed_dict=None,
burn_iters=2,
min_iters=10,
store_trace=False,
store_memory_usage=True,
name=None,
extras=None,
mbs=0
)
Выполняет операцию или тензор в заданной сессии. Сообщает результаты.
| Аргументы | |
|---|---|
sess | Session объект для использования в измерении времени. |
op_or_tensor | Operation или Tensor для бенчмарка. |
feed_dict | Список значений для подстановки для каждой итерации операции (см. параметр feed_dict в Session.run). |
burn_iters | Количество итераций разгона. |
min_iters | Минимальное количество итераций для измерения времени. |
store_trace | Булево значение, указывает, следует ли запустить дополнительную не измеряемую итерацию и сохранить трассировку итерации в возвращаемых дополнительных данных. Трассировка будет сохранена как строка в формате трассировки Google Chrome в поле extras "full_trace_chrome_format". Обратите внимание, что трассировка не будет сохранена в протоколе test_log_pb2.TestResults. |
store_memory_usage | Булево значение, указывает, следует ли запустить дополнительную не измеряемую итерацию, вычислить использование памяти и сохранить его в дополнительных полях. |
name | (необязательно) Переопределение имени BenchmarkEntry с помощью name. В противном случае оно выводится из имени метода верхнего уровня. |
extras | (необязательно) Словарь, сопоставляющий строковые ключи дополнительной информации о бенчмарке. Значения могут быть либо числами с плавающей точкой, либо значениями, которые могут быть преобразованы в строки. |
mbs | (необязательно) Количество мегабайтов, перемещенных этой операцией, используемое для расчета пропускной способности операции. |
| Возвращаемое значение | |
|---|---|
Словарь, содержащий пары ключ-значение, переданные в report_benchmark. Если используется опция store_trace, то full_chrome_trace_format будет включено в возвращаемый словарь, даже если оно не передано в report_benchmark с extras. |
© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/test/Benchmark