megengine.traced_module.traced_module.InternalGraph.replace_node

InternalGraph.replace_node(repl_dict)[源代码]

Replace the Nodes in the graph.

参数

repl_dict (Dict[Node, Node]) – the map {old_Node: new_Node} that specifies how to replace the Nodes.