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