Specialization

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

此類用於多圖編譯上下文，在該上下文中，我們會生成多個專門化的圖，並在執行時將其分派到合適的圖。這使我們能夠透過為常見模式（例如 x.shape[0] % 16 == 0）建立專門化的版本來最佳化效能和通用性之間的權衡，同時保持通用的回退機制。