strict_fusion

class torch.jit.strict_fusion

在推理中，如果沒有融合所有節點，或在訓練中沒有符號化區分，則會報錯。

示例：強制融合加法運算。

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