megengine.utils.network.Network.replace_vars

Network.replace_vars(repl_dict)[源代码]

Replaces vars in the graph.

参数

repl_dict (Dict[VarNode, VarNode]) – the map {old_var: new_var} that specifies how to replace the vars.

上一页

megengine.utils.network.Network.replace_oprs

下一页

megengine.utils.network.Network.reset_batch_size