openvinotoolkit/openvino

[Good First Issue]: support aten::tensordot

Open

#29.651 aberto em 24 de mar. de 2025

Ver no GitHub
 (6 comments) (0 reactions) (1 assignee)C++ (3.229 forks)auto 404
category: PyTorch FEgood first issueno_stale

Métricas do repositório

Stars
 (10.286 stars)
Métricas de merge de PR
 (Mesclagem média 14d 3h) (305 fundiu PRs em 30d)

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

Guia do colaborador

Direção de pesquisa
Estude as implementações existentes de operações aten no código fonte do OpenVINO (por exemplo, aten::matmul, aten::bmm) e a especificação PyTorch de aten::tensordot. Identifique a lógica de inferência de forma e kernel necessária.
Pilha de tecnologia
python
Domain
ai
Tipo Issue
Funcionalidade
DifficultyDificuldade de implementação estimada para um novo contribuidor, de 1 para alterações muito pequenas a 5 para trabalho de nível especializado.
3
Tempo estimadoUm intervalo de tempo aproximado para um colaborador experiente investigar, implementar, testar e preparar um pull request.
1-2 dias
Status da atividadeQuão disponível o issue aparece agora: novo, ativo, obsoleto, bloqueado ou aguardando entrada do mantenedor.
Ativo
ClarityCom que clareza o issue explica a mudança esperada, os critérios de aceitação e a próxima etapa.
Claro
Prerequisites
C++ programmingOpenVINO operation architecturePyTorch operator semantics
Simpatia para novatosUma pontuação de 1 a 100 que estima o quão acessível este issue é para colaboradores iniciantes.
60