严格融合

class torch.jit.strict_fusion

如果在推理中没有融合所有节点，或在训练中没有进行符号化微分，则给出错误。

示例： 强制融合加法。

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