strict_fusion

class torch.jit.strict_fusion

如果推理中并非所有节点都已融合，或者训练中并非所有节点都已符号化微分，则会给出错误。

示例：强制融合加法。

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