publiczna klasa końcowa ResourceSparseApplyMomentum
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 * pęd + grad var -= lr * accum
Klasy zagnieżdżone
| klasa | ResourceSparseApplyMomentum.Opcje | Opcjonalne atrybuty ResourceSparseApplyMomentum | |
Stałe
| Strunowy | OP_NAME | Nazwa tej operacji znana silnikowi rdzenia TensorFlow |
Metody publiczne
| static <T rozszerza TType > ResourceSparseApplyMomentum | |
| statyczny ResourceSparseApplyMomentum.Options | useLocking (boolowski useLocking) |
| statyczny ResourceSparseApplyMomentum.Options | 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: „ResourceSparseApplyMomentum”
Metody publiczne
public static ResourceSparseApplyMomentum create ( Zakres zakresu, Operand <?> var, Operand <?> 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ę ResourceSparseApplyMomentum.
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 |
Zwroty
- nową instancję ResourceSparseApplyMomentum
public static ResourceSparseApplyMomentum.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 ResourceSparseApplyMomentum.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. |
|---|