strict_fusion

class torch.jit.strict_fusion

当在推理时未能融合所有节点，或在训练时未能符号化微分时，会引发错误。

示例：强制融合加法操作。

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