openvinotoolkit/openvino

[Good First Issue]: support aten::tensordot

Open

#29 651 ouverte le 24 mars 2025

Voir sur GitHub
 (6 commentaires) (0 réactions) (1 assigné)C++ (3 229 forks)auto 404
category: PyTorch FEgood first issueno_stale

Métriques du dépôt

Stars
 (10 286 stars)
Métriques de merge PR
 (Merge moyen 14j 3h) (305 PRs mergées en 30 j)

Description

Request Description

Support for aten::tensordot in OpenVINO would enhance compatibility with PyTorch models by enabling tensor contraction through matrix multiplication followed by summation over specified axes. This operation is essential for various machine learning and numerical computing applications.

Feature Use Case

The tensordot operation is widely used in deep learning for tasks such as attention mechanisms, tensor decompositions, and high-dimensional matrix multiplications. Adding this functionality would improve inference optimization and ensure better integration with models requiring aten::tensordot.

Issue submission checklist

  • The feature request or improvement must be related to OpenVINO

Guide contributeur

Direction de recherche
Étudiez les implémentations existantes des opérations aten dans le code source d'OpenVINO (par exemple, aten::matmul, aten::bmm) et la spécification PyTorch de aten::tensordot. Identifiez la logique d'inférence de forme et de noyau requise.
Stack technique
python
Domaine
ai
Type d'issue
Fonctionnalité
DifficultéDifficulté estimée pour un nouveau contributeur, de 1 pour un très petit changement à 5 pour un travail expert.
3
Temps estiméFourchette de temps approximative pour investiguer, implémenter, tester et préparer une pull request.
1-2 jours
Statut d'activitéDisponibilité apparente de l'issue : fraîche, active, ancienne, bloquée ou en attente d'un mainteneur.
Active
ClartéClarté avec laquelle l'issue explique le changement attendu, les critères d'acceptation et la prochaine étape.
Claire
Prérequis
C++ programmingOpenVINO operation architecturePyTorch operator semantics
Accessibilité débutantScore de 1 à 100 estimant l'accessibilité de cette issue pour un premier contributeur.
60