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

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

用法:

eliminate_dead_code()

返回：

图表是否因通过而更改。

返回类型：

bool

根据每个节点的用户数量以及节点是否有任何副作用，从图中删除所有死代码。调用前必须对图进行拓扑排序。

例子：

在消除死代码之前，下面a = x + 1中的a没有用户，因此可以从图中消除而不会产生影响。

def forward(self, x):
    a = x + 1
    return x + self.attr_1

消除死代码后，a = x + 1 已被删除，剩下的 forward 仍然存在。

def forward(self, x):
    return x + self.attr_1

注意

保证此 API 的向后兼容性。