strict_fusion¶

class torch.jit.strict_fusion[來源][來源]¶

如果在推論中並非所有節點都已融合，或在訓練中以符號方式區分，則會產生錯誤。

範例：強制融合加法。

@torch.jit.script
def foo(x):
    with torch.jit.strict_fusion():
        return x + x + x

文件