Vue d'ensemble

PyTorch 2.1, publie le 5 octobre 2023, stabilise le support des formes dynamiques dans torch.compile et introduit torch.export pour l'export de modeles.

Fonctionnalites principales

Les formes dynamiques permettent a torch.compile de gerer des tenseurs de taille variable sans recompilation. torch.export produit un graphe IR portable.

python 
import torch

@torch.compile(dynamic=True)
def f(x):
    return x.sin() + x.cos()

print(f(torch.randn(4)))
print(f(torch.randn(8)))  # pas de recompilation

Sources