torch.fx.experimental.proxy_tensor.selective_decompose

torch.fx.experimental.proxy_tensor.selective_decompose(joint_gm, *args, decomposition, should_decompose, trace_joint_graph) [источник]

Повторно трассирует модуль совместного графа и выборочно применяет декомпозицию.