Add `min` and `max` rev specializations?

Aperta
#3,249 10 commenti 3 reazioni 0 assegnatari Vedi su GitHub

I maintainer di solito rispondono entro 1 giorno

Nessuno ha ancora preso questa issue.

Valutazione

Difficoltà
5/5
Tempo stimato
Più di una settimana
Idoneità per principianti
35/100
Tipo di issue
Funzionalità
Chiarezza
Da chiarire
Stato di attività
Ferma
Stack tecnologico
cpp
Ambito
backend

Direzione di ricerca

Leggi prima le discussioni collegate di Stack Exchange e PyTorch, quindi esamina le API min/max esistenti e i pattern di specializzazione reverse-mode nella libreria. Chiarisci il comportamento previsto per scalari e vettori, inclusi gli argomenti uguali e i gradienti proposti; il lavoro è completato quando le specializzazioni concordate sono implementate con test che coprono questi casi.

Scritto dal modello di indicizzazione a partire dal testo della issue.

Descrizione

good first issue

From this stackoverflow post it looks like we can phrase max(a,b) into

$$
max(a,b) = \frac{a + b + |a-b|}{2}
$$

and similarly for min

$$
min(a,b) = \frac{a + b - |a-b|}{2}
$$

So except for the case when $a=b$ this function is differentiable. For max we have

$$
\begin{align}
\frac{\partial max}{a} = \frac{1}{2} + \frac{sign(a-b)}{2}
\frac{\partial max}{a} = \frac{1}{2} + \frac{sign(a-b)}{2}
\end{align}
$$

so if a>b the gradient is 1 and if a<b the gradient is 1/2. For min we have

$$
\begin{align}
\frac{\partial min}{a} = \frac{1}{2} - \frac{sign(a-b)}{2}
\frac{\partial min}{a} = \frac{1}{2} - \frac{sign(a-b)}{2}
\end{align}
$$

where if $a>b$ the gradient is 0 and if $a<b$ the gradient is 1/2

From this it seems like pytorch does something similar to the above. I think this would be nice to implement. We can also write min(vector) for these as well.

Lingua principale
C++
Stelle
845
Fork
225
Merge medio
3g 9h
PR unite (30g)
12

Preparare l'ambiente

Come iniziare

  1. Leggi tutta la issue e poi la guida ai contributi del progetto.
  2. Commenta sulla issue per dire che te ne occupi tu — evita che due persone facciano lo stesso lavoro.
  3. Fai un fork del repository e lavora su un branch.
  4. Apri una pull request che faccia riferimento al numero della issue.

Altre issue di stan-dev/math

Tutte le issue di stan-dev/math

Issue simili

Altre issue su C++

Ricevi le nuove issue nella tua casella

Un breve riepilogo di issue GitHub adatte ai principianti.