Shortcuts

strict_fusion

class torch.jit.strict_fusion[source]

This class errors if not all nodes have been fused in inference, or symbolically differentiated in training.

Example:

Forcing fusion of additions.

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

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources