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 的向後兼容性。