Vue d'ensemble
PyTorch 1.9, publié le 16 juin 2021, stabilise le module torch.linalg pour l'algèbre linéaire et introduit l'interpréteur mobile pour le déploiement sur appareils embarqués.
Fonctionnalités principales
torch.linalg stable
Le module torch.linalg fournit des fonctions d'algèbre linéaire compatibles avec l'API NumPy : décomposition, inversion, résolution de systèmes linéaires, toutes différentiables.
python
import torch
A = torch.randn(3, 3)
b = torch.randn(3)
# Résolution de système linéaire Ax = b
x = torch.linalg.solve(A, b)
print(f'Solution : {x}')
# Décomposition en valeurs singulières
U, S, Vh = torch.linalg.svd(A)
print(f'Valeurs singulières : {S}')Interpréteur mobile
L'interpréteur mobile réduit la taille du runtime PyTorch pour le déploiement sur appareils mobiles. Il charge uniquement les opérateurs utilisés par le modèle, réduisant significativement l'empreinte.
python
import torch
from torch.utils.mobile_optimizer import optimize_for_mobile
# Préparation du modèle pour mobile
model = torch.jit.script(mon_modele)
optimized = optimize_for_mobile(model)
# Sauvegarde pour l'interpréteur mobile
optimized._save_for_lite_interpreter('model_mobile.ptl')
print('Modèle optimisé pour mobile sauvegardé')