openvinotoolkit/openvino

[Good First Issue]: support aten::tensordot

Open

#29.651 aperta il 24 mar 2025

Vedi su GitHub
 (6 commenti) (0 reazioni) (1 assegnatario)C++ (3229 fork)auto 404
category: PyTorch FEgood first issueno_stale

Metriche repository

Star
 (10.286 star)
Metriche merge PR
 (Merge medio 14g 3h) (305 PR mergiate in 30 g)

Descrizione

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

Guida contributor

Direzione di ricerca
Studia le implementazioni esistenti delle operazioni aten nel codice sorgente di OpenVINO (ad es. aten::matmul, aten::bmm) e la specifica PyTorch di aten::tensordot. Identifica la logica di inferenza della forma e del kernel richiesta.
Tech stack
python
Dominio
ai
Tipo issue
Funzionalità
DifficoltàQuanto dovrebbe essere impegnativa per un nuovo contributor.
3
Tempo stimatoTempo stimato per completare e verificare uno scope piccolo.
1-2 giorni
Stato attivitàQuanto è probabile che la issue sia ancora attiva e reviewable.
Attiva
ChiarezzaQuanto chiaramente la issue descrive problema, scope e risultato atteso.
Chiara
Prerequisiti
C++ programmingOpenVINO operation architecturePyTorch operator semantics
Adatta ai principiantiQuanto la issue è adatta a contributor alla prima esperienza.
60