Python PyTorch Graph.inserting_before用法及代碼示例

本文簡要介紹python語言中 torch.fx.Graph.inserting_before 的用法。

用法:

inserting_before(n=None)

設置create_node 和伴隨方法將插入圖表的點。

在‘with’ 語句中使用時，這將臨時設置插入點，然後在 with 語句退出時恢複它：

with g.inserting_before(n):
    ... # inserting before node n
... # insert point restored to what it was previously
g.inserting_before(n) #  set the insert point permanently

參數：

n(可選[節點])：要在其之前插入的節點。如果沒有，這將插入之前

整個圖形的開始。

返回：

將恢複 __exit__ 上的插入點的資源管理器。

注意

保證此 API 的向後兼容性。