torch.fx.experimental.optimization.extract_subgraph

torch.fx.experimental.optimization.extract_subgraph(orig_module, nodes, inputs, outputs) [исходный код]

Принимает списки узлов существующего графа, образующих подграф, и возвращает подмодуль, выполняющий этот подграф.