publiczna klasa końcowa SparseApplyMomentum
Zaktualizuj odpowiednie wpisy w „*var” i „*accum” zgodnie ze schematem pędu.
Ustaw use_nesterov = True, jeśli chcesz użyć pędu Niestierowa.
Oznacza to, że dla wierszy, dla których mamy grad, aktualizujemy var i accum w następujący sposób:
$$accum = accum * momentum + grad$$$$var -= lr * accum$$
Klasy zagnieżdżone
| klasa | Opcje SparseApplyMomentum | Opcjonalne atrybuty dla SparseApplyMomentum | |
Stałe
| Smyczkowy | OP_NAME | Nazwa tej operacji znana silnikowi rdzenia TensorFlow |
Metody publiczne
| Wyjście <T> | jako wyjście () Zwraca symboliczny uchwyt tensora. |
| statyczny <T rozszerza TType > SparseApplyMomentum <T> | |
| Wyjście <T> | na zewnątrz () To samo co „var”. |
| statyczny SparseApplyMomentum.Opcje | useLocking (boolowski useLocking) |
| statyczny SparseApplyMomentum.Opcje | useNesterov (Boolean useNesterov) |
Metody dziedziczone
Stałe
publiczny statyczny końcowy ciąg znaków OP_NAME
Nazwa tej operacji znana silnikowi rdzenia TensorFlow
Wartość stała: „SparseApplyMomentum”
Metody publiczne
publiczne wyjście <T> asOutput ()
Zwraca symboliczny uchwyt tensora.
Dane wejściowe operacji TensorFlow są wynikami innej operacji TensorFlow. Ta metoda służy do uzyskania symbolicznego uchwytu reprezentującego obliczenia danych wejściowych.
public static SparseApplyMomentum <T> create (Zakres zakresu , Operand <T> var, Operand <T> accum, Operand <T> lr, Operand <T> grad, Operand <? rozszerza TNumber > indeksy, Operand <T> pęd, Opcje ... opcje)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację SparseApplyMomentum.
Parametry
| zakres | aktualny zakres |
|---|---|
| odm | Powinno pochodzić ze zmiennej (). |
| gromadzić | Powinno pochodzić ze zmiennej (). |
| lr | Szybkość uczenia się. Musi być skalarem. |
| absolwent | Gradient. |
| indeksy | Wektor indeksów do pierwszego wymiaru var i accum. |
| pęd | Pęd. Musi być skalarem. |
| opcje | przenosi opcjonalne wartości atrybutów |
Powroty
- nowa instancja SparseApplyMomentum
public static SparseApplyMomentum.Options useLocking (boolean useLocking)
Parametry
| użyjBlokowanie | Jeśli `True`, aktualizacja tensorów var i accum będzie zabezpieczona blokadą; w przeciwnym razie zachowanie jest niezdefiniowane, ale może wykazywać mniejszą rywalizację. |
|---|
public static SparseApplyMomentum.Options useNesterov (Boolean useNesterov)
Parametry
| użyj Niestierowa | Jeśli „True”, tensor przekazany do obliczenia grad będzie miał postać var - lr * pęd * accum, więc ostatecznie otrzymana wartość var to w rzeczywistości var - lr * pęd * accum. |
|---|