strict_fusion

class torch.jit.strict_fusion

如果在推理中並非所有節點都已融合，或在訓練中並非所有節點都已符號化微分，則報錯。

示例：強制加法融合。

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