Python PyTorch Graph.node_copy用法及代码示例

本文简要介绍python语言中 torch.fx.Graph.node_copy 的用法。

用法:

node_copy(node, arg_transform=<function Graph.<lambda>>)

参数：

• node(Node) -要复制到 self 的节点。

• arg_transform(可调用[[Node],参数]) - 转换的函数Node节点中的参数args和kwargs到等价的论点中self.在最简单的情况下，这应该从原始图中的节点映射表中检索一个值到self.

将节点从一个图形复制到另一个图形。 arg_transform 需要将参数从节点图转换为自身图。例子：

# Copying all the nodes in `g` into `new_graph`
g : torch.fx.Graph = ...
new_graph = torch.fx.graph()
value_remap = {}
for node in g.nodes:
    value_remap[node] = new_graph.node_copy(node, lambda n : value_remap[n])

注意

保证此 API 的向后兼容性。