Specialization

class torch.fx.experimental.symbolic_shapes.Specialization(source, check_fn)

此类用于多图编译上下文，在该上下文中，我们会生成多个专门化的图，并在运行时将其分派到合适的图。这使我们能够通过为常见模式（例如 x.shape[0] % 16 == 0）创建专门化的版本来优化性能和通用性之间的权衡，同时保持通用的回退机制。