分析器

分析器用于查找层级依赖关系以识别连接的符号。

函数

analyze_symbols

追踪层依赖并返回一个有效的符号映射。

analyze_symbols(model, concrete_args=None, _strict=True)

追踪层依赖并返回一个有效的符号映射。

跟踪过程是原地进行的。

Parameters:
  • model (Module) – 要跟踪的模型。

  • concrete_args (Dict[str, Any] | None) – [仅用于调试!] 传递给符号追踪器的模型.forward()的具体参数。

  • _strict (bool) – [仅用于调试!] 是否强制执行严格的依赖跟踪。

Return type:

SymMap