RFC: add support for batched matmul / einsum / batched tensordot

Abierto
#727 21 comentarios 2 reacciones 0 asignados Ver en GitHub

Nadie ha tomado este issue todavía.

Evaluación

Dificultad
5/5
Tiempo estimado
Más de una semana
Aptitud para principiantes
25/100
Tipo de issue
Nueva funcionalidad
Claridad
Necesita aclaración
Estado de actividad
Estancado
Área
data

Línea de trabajo

El issue no indica archivos, tests ni puntos de entrada. Empieza revisando el estándar de Array API y comparando los enfoques propuestos batched_matmul, batched tensordot y einsum; se considera terminado cuando se selecciona y especifica una operación compatible, en lugar de dejar abiertas varias alternativas.

Escrito por el modelo de indexación a partir del texto del issue.

Descripción

Needs Discussion RFC topic: Linear Algebra

Currently I don't see a way to implement einsum or batched matmul using the standard. There is a way with explicit loop, but that does not count.

Some ways this can be addressed:

  • minimal implementation would include batched_matmul (signature bij, bjk -> bik). Other operations can be reduced to it with reshapes. This may include additional copies
  • modification of tensordot to include batched dimensions thus should be more efficient, but AFAIK this method is not surfaced in any major framework
  • include einsum in the standard
Lenguaje dominante
Python
Estrellas
281
Forks
52
Sin PR fusionados en 30 d

Guía de contribución

Abrir la guía de contribución

Primeros pasos

  1. Lee el issue completo y luego la guía de contribución del proyecto.
  2. Comenta en el issue que vas a ocuparte — evita que dos personas hagan lo mismo.
  3. Haz un fork del repositorio y trabaja en una rama.
  4. Abre un pull request que haga referencia al número del issue.

Más de data-apis/array-api

Todos los issues de data-apis/array-api

Issues similares

Más issues de Python

Recibe los nuevos issues en tu correo

Un resumen breve de issues de GitHub para principiantes.