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
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. |
---|