publiczna klasa końcowa ResourceApplyMomentum
Zaktualizuj „*var” zgodnie ze schematem pędu.
Ustaw use_nesterov = True, jeśli chcesz użyć pędu Niestierowa.
accum = accum * pęd + grad var -= lr * accum
Klasy zagnieżdżone
| klasa | ResourceApplyMomentum.Opcje | Opcjonalne atrybuty ResourceApplyMomentum | |
Stałe
| Strunowy | OP_NAME | Nazwa tej operacji znana silnikowi rdzenia TensorFlow |
Metody publiczne
| statyczny <T rozszerza TType > ResourceApplyMomentum | |
| statyczny ResourceApplyMomentum.Options | useLocking (boolowski useLocking) |
| statyczny ResourceApplyMomentum.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: „ResourceApplyMomentum”
Metody publiczne
public static ResourceApplyMomentum create ( Zakres zasięgu , Operand <?> var, Operand <?> accum, Operand <T> lr, Operand <T> grad, Operand <T> pęd, Opcje... opcje)
Metoda fabryczna służąca do tworzenia klasy opakowującej nową operację ResourceApplyMomentum.
Parametry
| zakres | aktualny zakres |
|---|---|
| odm | Powinno pochodzić ze zmiennej (). |
| gromadzić | Powinno pochodzić ze zmiennej (). |
| lr | Współczynnik skalowania. Musi być skalarem. |
| absolwent | Gradient. |
| pęd | Pęd. Musi być skalarem. |
| opcje | przenosi opcjonalne wartości atrybutów |
Zwroty
- nowa instancja ResourceApplyMomentum
public static ResourceApplyMomentum.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 ResourceApplyMomentum.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. |
|---|