Специализация

class torch.fx.experimental.symbolic_shapes.Specialization(source, check_fn) [исходный код]

Этот класс используется в контекстах компиляции с несколькими графами, в которых мы создаём несколько специализированных графов и во время выполнения направляем вызов к соответствующему графу. Это позволяет оптимизировать компромисс между производительностью и универсальностью, создавая специализированные версии для распространённых шаблонов (например, x.shape[0] % 16 == 0) и сохраняя общий резервный вариант.