什么是 torch.compile？ Dynamo、PyTorch 和更快的 AI 模型解释

描述

您有一个可以正确运行的 PyTorch 模型，并且您希望让它运行得更快。您添加一行模型等于 torch.compile。第一次运行可能需要更长的时间，因为 PyTorch 必须编译代码。之后，模型可能会运行得更快。但 PyTorch 到底在做什么呢？以及为什么……