publiczna klasa końcowa ResourceApplyKerasMomentum
Zaktualizuj „*var” zgodnie ze schematem pędu.
Ustaw use_nesterov = True, jeśli chcesz użyć pędu Niestierowa.
accum = accum * pęd - lr * grad var += accum
Klasy zagnieżdżone
klasa | ResourceApplyKerasMomentum.Opcje | Opcjonalne atrybuty ResourceApplyKerasMomentum |
Stałe
Strunowy | OP_NAME | Nazwa tej operacji znana silnikowi rdzenia TensorFlow |
Metody publiczne
static <T rozszerza TType > ResourceApplyKerasMomentum | |
statyczny ResourceApplyKerasMomentum.Options | useLocking (boolowski useLocking) |
statyczny ResourceApplyKerasMomentum.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: „ResourceApplyKerasMomentum”
Metody publiczne
public static ResourceApplyKerasMomentum create ( Zakres zakresu, 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ę ResourceApplyKerasMomentum.
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 ResourceApplyKerasMomentum
publiczny statyczny ResourceApplyKerasMomentum.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 ResourceApplyKerasMomentum.Options useNesterov (Boolean useNesterov)
Parametry
użyj Niestierowa | Jeśli „True”, tensor przekazany do obliczenia grad będzie miał postać var + pęd * accum, więc ostatecznie otrzymana wartość var będzie wynosić w rzeczywistości var + pęd * accum. |
---|