EngineInspector

class tensorrt.EngineInspector

Engine Inspector 用于打印引擎或执行上下文的层信息。在调用 get_layer_information() 或 get_engine_information() 之前，必须设置引擎或上下文。

打印信息的量取决于构建引擎时构建器配置文件的性能分析详细程度设置。默认情况下，性能分析详细程度设置为 ProfilingVerbosity.LAYER_NAMES_ONLY，并且仅打印层名称。如果性能分析详细程度设置为 ProfilingVerbosity.DETAILED，则将打印层名称和层参数。如果性能分析详细程度设置为 ProfilingVerbosity.NONE，则不打印任何层信息。

变量:

• engine – ICudaEngine 设置或获取当前正在检查的引擎。

• context – IExecutionContext 设置或获取当前正在检查的上下文。

• error_recorder – IErrorRecorder 应用程序实现的 TensorRT 对象错误报告接口。

__init__(*args, **kwargs)

get_engine_information(self: tensorrt.tensorrt.EngineInspector, format: tensorrt.tensorrt.LayerInformationFormat) → str

获取一个字符串，描述当前引擎或执行上下文中所有层的信息。

参数:

format – LayerInformationFormat 层信息应打印成的格式。

返回:

一个字符串，描述当前引擎或执行上下文中所有层的信息。

get_layer_information(self: tensorrt.tensorrt.EngineInspector, layer_index: int, format: tensorrt.tensorrt.LayerInformationFormat) → str

获取一个字符串，描述当前引擎或执行上下文中特定层的信息。

参数:

• layer_index – 层的索引。它必须在 [0, engine.num_layers] 范围内。

• format – LayerInformationFormat 层信息应打印成的格式。

返回:

一个字符串，描述当前引擎或执行上下文中特定层的信息。