模块

模块

modelopt.torch.trace.modules.concat

支持追踪连接操作的模块。

modelopt.torch.trace.modules.nn

包含原生 torch 层特定符号信息和实现的模块。

包含特定于层的符号信息和实现的模块。