openvinotoolkit/openvino

[Good First Issue]: support aten::tensordot

Open

#29.651 geöffnet am 24. März 2025

Auf GitHub ansehen
 (6 Kommentare) (0 Reaktionen) (1 zugewiesene Person)C++ (3.229 Forks)auto 404
category: PyTorch FEgood first issueno_stale

Repository-Metriken

Stars
 (10.286 Stars)
PR-Merge-Metriken
 (Durchschn. Merge 14T 3h) (305 gemergte PRs in 30 T)

Beschreibung

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

Contributor Guide

Research-Richtung
Untersuche bestehende Implementierungen von aten Operationen im OpenVINO Quellcode (z. B. aten::matmul, aten::bmm) und die PyTorch Spezifikation für aten::tensordot. Ermittle die erforderliche Forminferenz und Kernel Logik.
Tech Stack
python
Domain
ai
Issue Type
Funktion
SchwierigkeitGeschätzte Implementierungsschwierigkeit für neue Contributors, von 1 für sehr kleine Änderungen bis 5 für Expertenarbeit.
3
Geschätzte ZeitEin grober Zeitrahmen, um zu recherchieren, zu implementieren, zu testen und einen Pull Request vorzubereiten.
1-2 Tage
AktivitätsstatusWie verfügbar das Issue gerade wirkt: frisch, aktiv, stale, blockiert oder wartet auf Maintainer-Input.
Aktiv
KlarheitWie klar das Issue die erwartete Änderung, Akzeptanzkriterien und den nächsten Schritt erklärt.
Klar
Voraussetzungen
C++ programmingOpenVINO operation architecturePyTorch operator semantics
EinsteigerfreundlichkeitEin Score von 1 bis 100, der schätzt, wie zugänglich dieses Issue für First-Time Contributors ist.
60